George Hotz 643cbdfd50 make embedding and GPT-2 fast (#1631) ... * make embedding fast * jit more, variable shape support * print mem bw		2023-08-22 15:14:38 -07:00
..
dist	distributed collectives (#1519)	2023-08-11 10:22:07 -07:00
external_copy_benchmark.py	Symbolic Shape JIT main PR (#1353)	2023-08-18 14:39:55 -07:00
external_hlb_cifar.py	Fix naming conflict with huggingface datasets (#1161)	2023-07-07 10:43:44 -07:00
external_llama_eval.py	Add option in llama.py to quantize weights to int8 at runtime (#1289)	2023-07-24 17:22:38 -07:00
external_metal_uaf.py	fix metal uaf (#964)	2023-06-09 21:28:06 -07:00
external_model_benchmark.py	single model (#1560)	2023-08-16 13:22:19 -07:00
external_multi_gpu.py	helpers.py: improved test coverage + exception handling (#1165)	2023-07-07 10:26:05 -07:00
external_osx_profiling.py	multi cl_queue (#762)	2023-05-03 12:15:28 -07:00
external_test_allocator_on_models.py	move state to nn/state (#1619)	2023-08-22 07:36:24 -07:00
external_test_embedding.py	make embedding and GPT-2 fast (#1631)	2023-08-22 15:14:38 -07:00
external_test_gpu_ast.py	move to shapetracker.py	2023-03-11 07:50:07 -08:00
external_test_image.py	Revert "update editorconfig, enforce via CI (#1343)" (#1380)	2023-07-31 10:35:50 -07:00
external_test_llvm.py	fix up external tests	2023-03-06 06:52:28 -08:00
external_test_onnx_backend.py	exclude a few more onnx tests	2023-06-19 08:51:29 -07:00
external_test_opt.py	move state to nn/state (#1619)	2023-08-22 07:36:24 -07:00
external_test_optim.py	Add LAMB optimizer (#821)	2023-05-28 15:09:05 -07:00
external_test_speed_llama.py	move state to nn/state (#1619)	2023-08-22 07:36:24 -07:00
external_test_yolo.py	test cleanups	2023-03-03 06:36:06 -08:00
external_test_yolov8.py	move state to nn/state (#1619)	2023-08-22 07:36:24 -07:00
fuzz_shapetracker.py	Add Fuzz Test symbolic / shapetracker to CI. (#1278)	2023-07-19 09:05:45 -07:00
fuzz_symbolic.py	symbolic stride (#1326)	2023-07-23 12:41:22 -07:00
graph_batchnorm.py	move state to nn/state (#1619)	2023-08-22 07:36:24 -07:00