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
..
mlperf	feat: add train scaffolding (#859)	2023-05-30 07:10:40 -07:00
yolo	Permute examples (#731)	2023-03-29 05:07:06 +04:00
__init__.py	failing llama test	2023-03-11 16:28:10 -08:00
benchmark_train_efficientnet.py	rename log_softmax, support dim, fix onnx Softmax	2023-02-24 10:11:24 -08:00
compile_efficientnet.py	Devicebufferless (#708)	2023-03-18 14:40:23 -07:00
compile_tensorflow.py	moved extras/jit.py -> tinygrad/jit.py (#599)	2023-02-25 08:32:33 -08:00
deep_deterministic_policy_gradient.py	Fixed DDPG example (#667)	2023-03-09 11:49:52 -08:00
efficientnet.py	Cast image to float32 (#702)	2023-03-14 08:13:19 -07:00
hlb_cifar10.py	nn init matches torch (#901)	2023-06-01 21:24:11 -07:00
hlb_cifar10_torch.py	nn init matches torch (#901)	2023-06-01 21:24:11 -07:00
llama.py	safetensors! (#903)	2023-06-02 13:41:09 -07:00
mnist_gan.py	Cast Tensor data to float32 (#703)	2023-03-14 23:09:41 -07:00
serious_mnist.py	rename log_softmax, support dim, fix onnx Softmax	2023-02-24 10:11:24 -08:00
stable_diffusion.py	start on mlperf models	2023-05-10 16:30:49 -07:00
train_efficientnet.py	train_efficientnet: only import datasets.imagenet when IMAGENET is set (#899)	2023-06-01 19:19:52 -07:00
train_resnet.py	ResNet: match implementation with Nvidia and PyTorch (#770)	2023-05-10 09:01:22 -07:00
transformer.py	add int64 as supported dtype from numpy (#699)	2023-03-18 17:15:04 -07:00
vgg7.py	Remove Tensor.data (#565)	2023-02-18 16:36:12 -08:00
vit.py	Remove Tensor.data (#565)	2023-02-18 16:36:12 -08:00
yolov3.py	Permute examples (#731)	2023-03-29 05:07:06 +04:00
yolov8.py	yolov8 works up to the MaxPool	2023-02-22 19:32:13 -08:00