George Hotz d58586bb17 safetensors! (#903) ...

* safetensors test

* safe_save

* load back with real safetensors

* bugfix in device name. add simple torch_load

* it works for llama, but it's slower...

* mmap

* no intermediate

* load mmaped

* readinto speed

* not ready yet

* revert that

2023-06-02 13:41:09 -07:00

..

external

Onnx ops And, Or, Xor, Not (#847)

2023-05-29 11:09:20 -07:00

extra

make tests faster + add onnx (#815)

2023-05-27 08:53:32 -07:00

models

Zero dim Tensor support (#777)

2023-06-01 11:32:02 -07:00

unit

safetensors! (#903)

2023-06-02 13:41:09 -07: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

Add conv binops_no_rerun test assertions (#665)

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

test_custom_function.py

Disktensors! (#819)

2023-05-28 15:40:37 -07:00

test_dtype.py

Llvm int support (#866)

2023-05-30 17:49:26 -07:00

test_gc.py

fix up eye and fix gc test

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

test_jit.py

assertions for jit

2023-05-05 21:56:32 -07:00

test_net_speed.py

test touchups

2023-05-26 02:39:42 +00:00

test_nn.py

support Int64 type in cstyle gen (#860)

2023-05-30 16:04:46 -07:00

test_ops.py

Fix .std() tests on torch=1.13 (#904)

2023-06-02 07:33:51 -07:00

test_optim.py

Add weight decay to SGD (#883)

2023-06-01 13:13:18 -07:00

test_randomness.py

add nn layer tests

2023-06-01 21:34:24 -07:00

test_specific_conv.py

disable that test on LLVM. i have to stop pushing to master

2023-05-26 03:11:03 +00:00

test_speed_v_torch.py

good changes from the M1 Tensor Core project (#730)

2023-03-29 05:11:02 +04:00

test_tensor.py

Zero dim Tensor support (#777)

2023-06-01 11:32:02 -07:00