George Hotz 036737a12a mem_estimate tracks bytes, not items

2023-03-10 09:44:12 -08:00

..

external

add challenge tests

2023-03-07 19:39:04 -08:00

models

ConvNeXt + extras (#657)

2023-03-06 22:10:56 -08:00

unit

int32, and refactor pad/shrink

2023-03-09 12:57:17 -08:00

__init__.py

All devices are equal! (#196)

2020-12-15 23:44:08 -08:00

Dockerfile

testdocker

2023-01-09 12:41:52 -08:00

test_assign.py

assign buffer reuse (#547)

2023-02-09 11:53:02 -06:00

test_conv_shapetracker.py

speed like you wouldn't believe (#626)

2023-03-02 07:49:19 -08:00

test_conv.py

Fold reduce (#362)

2022-07-19 09:24:02 -07:00

test_custom_function.py

fix GPU import error and old python Tuple

2023-03-08 12:22:11 -08:00

test_example.py

fix CUDA float4 issues

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

test_gc.py

fix up eye and fix gc test

2023-02-27 06:53:18 -08:00

test_jit.py

moved extras/jit.py -> tinygrad/jit.py (#599)

2023-02-25 08:32:33 -08:00

test_net_speed.py

rename log_softmax, support dim, fix onnx Softmax

2023-02-24 10:11:24 -08:00

test_nn.py

relax atol, merge_view

2023-03-03 07:48:44 -08:00

test_ops.py

good changes from llama branch (#671)

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

test_optim.py

rename log_softmax, support dim, fix onnx Softmax

2023-02-24 10:11:24 -08:00

test_randomness.py

Add tests for randomness (#621)

2023-03-01 15:39:20 -08:00

test_specific_conv.py

clean up opt (#649)

2023-03-05 20:49:36 -08:00

test_speed_v_torch.py

mem_estimate tracks bytes, not items

2023-03-10 09:44:12 -08:00

test_tensor.py

ConvNeXt + extras (#657)

2023-03-06 22:10:56 -08:00