geohot 803b0aef28 track memory for numpy/torch

2023-03-11 20:39:10 -08:00

..

gemm

good changes from llama branch (#671)

2023-03-09 20:51:22 -08:00

triton

A Triton backend for tinygrad (#470)

2023-02-01 11:53:57 -08: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

more llama stuff

2023-03-10 10:48:10 -08:00

introspection.py

Refactor ASTs (#622)

2023-03-01 18:57:29 -08: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

onnx_ops.py

Eq magic function support (#683)

2023-03-11 10:31:46 -08:00

onnx.py

fine, openpilot can have 197 kernels

2023-02-27 11:48:36 -08:00

thneed.py

dtypes nice and clean (#673)

2023-03-10 16:56:07 -08:00

training.py

Remove Tensor.data (#565)

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

utils.py

track memory for numpy/torch

2023-03-11 20:39:10 -08:00