tinygrad/examples
George Hotz ea5dddc537
reduce collapse generic (#10045)
* reduce collapse generic

* new arange folder

* new range folding

* correct with sym

* all tests pass

* indexing ops passes

* failing tests

* fix tests, remove unused

* revert that

* torch indexing is fast

* skip on webgpu

* touchups

* comments
2025-04-26 09:13:24 -04:00
..
conversation_data Whisper + LLAMA + VITS (#2332) 2023-12-02 15:03:46 -08:00
llm.c CLANG -> CPU (#9189) 2025-02-20 18:03:09 -05:00
mlperf lint mlperf model_train (#10038) 2025-04-24 16:19:44 -04:00
openpilot add onnx frontend stub [pr] (#9558) 2025-03-24 12:24:34 +08:00
other_mnist reduce collapse generic (#10045) 2025-04-26 09:13:24 -04:00
rl more beautiful_cartpole with exposed hparams 2024-01-07 17:41:09 -08:00
sovits_helpers combine pad2d with pad (#7677) 2024-11-14 17:56:02 +08:00
tinychat tinychat in browser, Part 3: browser app (#9276) 2025-03-07 15:07:33 +08:00
vgg7_helpers leakyrelu to leaky_relu (#9270) 2025-02-26 13:22:08 -05:00
webgpu More yolo processing in tinygrad (#9928) 2025-04-24 16:21:46 -04:00
__init__.py failing llama test 2023-03-11 16:28:10 -08:00
beautiful_cartpole.py tinytqdm.set_description and tinytrange (#5101) 2024-06-22 14:45:06 -04:00
beautiful_cifar.py Fix mypy examples/beautiful_*.py (#6978) 2024-10-10 11:34:29 -04:00
beautiful_mnist.py Revert "switch beautiful_mnist to use new optimizer [pr] (#8231)" (#8233) 2024-12-13 19:07:09 -08:00
beautiful_mnist_multigpu.py Fix mypy examples/beautiful_*.py (#6978) 2024-10-10 11:34:29 -04:00
benchmark_onnx.py more stuff from DSP (#9689) 2025-04-02 15:27:48 +08:00
coder.py apply the same fix_bf16 in llama and coder (#3789) 2024-03-17 21:25:24 -04:00
compile_efficientnet.py CLANG -> CPU (#9189) 2025-02-20 18:03:09 -05:00
compile_tensorflow.py add onnx frontend stub [pr] (#9558) 2025-03-24 12:24:34 +08:00
conversation.py Fix examples/conversation.py (#8425) 2024-12-26 12:45:19 -05:00
efficientnet.py remove clang program header (#4422) 2024-05-04 08:38:01 -07:00
flux1.py flux set model path in args (#7660) 2024-11-12 22:11:40 -05:00
flux1_seed0.png Flux.1 (#6334) 2024-09-24 10:08:04 +08:00
gpt2.py Off by one error in start_pos (#9792) 2025-04-15 15:07:13 -04:00
handcode_opt.py hand_coded_optimizations returns list[Opt] [pr] (#9938) 2025-04-19 20:26:59 -04:00
hlb_cifar10.py MultiLazyBuffer is UOp [pr] (#8662) 2025-01-24 13:28:55 +09:00
llama.py validate llama quantize output (#7901) 2024-11-25 16:46:23 -05:00
llama3.py acc_dtype -> dtype (#9402) 2025-03-10 16:05:30 -04:00
mamba.py prev speed improvements (#5252) 2024-07-03 09:06:01 -07:00
mask_rcnn.py change Tensor.stack to method (#4719) 2024-05-24 17:04:19 -04:00
mixtral.py tinytqdm.set_description and tinytrange (#5101) 2024-06-22 14:45:06 -04:00
mnist_gan.py leakyrelu to leaky_relu (#9270) 2025-02-26 13:22:08 -05:00
olmoe.py olmoe memory usage cleanups 2025-03-19 12:28:18 +08:00
openelm.py nn.RMSNorm (#5272) 2024-07-02 21:39:01 -04:00
qwq.py QwQ-32B-Preview support (#7962) 2024-12-04 21:46:37 -05:00
sdv2.py Stable Diffusion v2 Inference (#5283) 2024-07-03 22:47:10 -04:00
sdxl.py GlobalCounters.reset() in sdxl step [pr] (#8664) 2025-01-17 21:10:28 -05:00
sdxl_seed0.png default threefry (#6116) 2024-09-25 17:45:13 +08:00
self_tokenize.py make self_tokenize output more like a python file (#8411) 2024-12-25 14:16:30 -05:00
serious_mnist.py combine pad2d with pad (#7677) 2024-11-14 17:56:02 +08:00
simple_conv_bn.py fix various examples (#4691) 2024-05-22 20:43:21 -04:00
so_vits_svc.py use tuple in isinstance for type checking (#9583) 2025-03-26 19:36:48 +08:00
stable_diffusion.py Remove wgpu specific checks from stable diffusion example (#7991) 2024-12-02 11:31:14 +01:00
stable_diffusion_seed0.png default threefry (#6116) 2024-09-25 17:45:13 +08:00
stunning_mnist.py stunning_mnist [run_process_replay] (#6828) 2024-10-01 15:00:48 +08:00
test_onnx_imagenet.py fixes from the dsp branch + 12500 lines (#9683) 2025-04-02 13:07:17 +08:00
test_pkl_imagenet.py more stuff from DSP (#9689) 2025-04-02 15:27:48 +08:00
torch_cuda_kernel.py hotfix: interop example (#9237) 2025-02-25 10:32:00 +03:00
train_efficientnet.py tinytqdm.set_description and tinytrange (#5101) 2024-06-22 14:45:06 -04:00
train_resnet.py move things, clean up extra (#2292) 2023-11-13 20:18:40 -08:00
transformer.py fixing transformer training bug (#9877) 2025-04-13 19:34:20 -04:00
vgg7.py waifu2x vgg7: testcase, auto-RGBA->RGB, function to grab pretrained models, training "fix" (#2117) 2023-10-19 22:07:15 -07:00
vit.py move to new cached fetch (#2493) 2023-11-28 17:36:55 -08:00
vits.py leakyrelu to leaky_relu (#9270) 2025-02-26 13:22:08 -05:00
whisper.py enable whisper batch for long sequences (#6458) 2024-09-17 00:42:10 -04:00
yolov3.py leakyrelu to leaky_relu (#9270) 2025-02-26 13:22:08 -05:00
yolov8-onnx.py add onnx frontend stub [pr] (#9558) 2025-03-24 12:24:34 +08:00
yolov8.py More yolo processing in tinygrad (#9928) 2025-04-24 16:21:46 -04:00