Mehmet Kuzucu c3173ff281 Add return statement to the train function (#1135) ...

add a return statement to the train function in order to provide access to the losses and accuracies lists

2023-07-05 08:13:38 -07:00

..

accel

move accel into extra

2023-06-23 16:38:15 -07:00

disk

Fast DiskTensor to other Tensor (#916)

2023-06-03 12:25:41 -07:00

gemm

hip matmul : add K support

2023-06-28 19:54:33 +00:00

intel

intel benchmark matmul gets 60 TFLOPS?

2023-06-04 17:01:50 +00:00

ptx

ptx example

2023-05-26 19:28:51 -07:00

rocm

faster RDNA assembly backend (#990)

2023-06-16 12:06:38 -07:00

triton

A Triton backend for tinygrad (#470)

2023-02-01 11:53:57 -08:00

archprobe.py

no werror in archprobe

2023-05-03 19:34:17 +00:00

augment.py

[Fix] fix typo: test_mnist -> datasets (#492)

2023-01-29 21:30:47 -08:00

gradcheck.py

Remove Tensor.data (#565)

2023-02-18 16:36:12 -08:00

helpers.py

ignore cloudpickle type for local mypy

2023-07-04 13:51:20 -07:00

hip_wrapper.py

hip matmul

2023-06-28 19:21:01 +00:00

introspection.py

Devicebufferless (#708)

2023-03-18 14:40:23 -07:00

kernel_search.py

move to shapetracker.py

2023-03-11 07:50:07 -08:00

lib_test_ast.py

fix ast test

2023-03-04 07:49:25 -08:00

lr_scheduler.py

LR Schedulers (#755)

2023-05-27 07:47:49 -07:00

onnx_ops.py

Handle broadcast flag on gemm (#1103)

2023-07-02 22:15:07 -07:00

onnx.py

Use generators instead of lists in anys and alls (#1111)

2023-07-03 16:06:06 -07:00

thneed.py

include missing queue in thneed load (#1095)

2023-07-02 12:33:59 -07:00

training.py

Add return statement to the train function (#1135)

2023-07-05 08:13:38 -07:00

utils.py

Windows & Ubuntu CLANG CI support (#1011)

2023-06-19 09:33:24 -07:00