George Hotz 0b733ba75e multi device training with GPT2 [pr] (#10375) ...

* multi device training with GPT2 [pr]

* Update grouper.py

2025-05-17 15:33:56 -07:00

..

conversation_data

Whisper + LLAMA + VITS (#2332)

2023-12-02 15:03:46 -08:00

llm.c

multi device training with GPT2 [pr] (#10375)

2025-05-17 15:33:56 -07:00

mlperf

move benchmark stat tracking to influxdb (#10185)

2025-05-15 16:14:56 -07: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

Fix yolo load failing silently (#10046)

2025-05-07 11:46:09 -07: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_multigpu.py

Fix mypy examples/beautiful_*.py (#6978)

2024-10-10 11:34:29 -04:00

beautiful_mnist.py

update dead link (#10242)

2025-05-09 19:59:52 -04:00

benchmark_onnx.py

more stuff from DSP (#9689)

2025-04-02 15:27:48 +08:00

coder.py

don't pass model in convert_from_huggingface and convert_from_gguf (#10094)

2025-04-28 20:11:19 -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_seed0.png

Flux.1 (#6334)

2024-09-24 10:08:04 +08:00

flux1.py

flux set model path in args (#7660)

2024-11-12 22:11:40 -05:00

gpt2.py

move benchmark stat tracking to influxdb (#10185)

2025-05-15 16:14:56 -07:00

handcode_opt.py

hand_coded_optimizations returns list[Opt] [pr] (#9938)

2025-04-19 20:26:59 -04:00

hlb_cifar10.py

clean: double import (#10345)

2025-05-15 20:15:09 -07:00

llama3.py

move benchmark stat tracking to influxdb (#10185)

2025-05-15 16:14:56 -07:00

llama.py

move benchmark stat tracking to influxdb (#10185)

2025-05-15 16:14:56 -07: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

minrf.py

randn_like in minrf (#10298)

2025-05-14 07:59:50 -04:00

mixtral.py

move benchmark stat tracking to influxdb (#10185)

2025-05-15 16:14:56 -07:00

mnist_gan.py

leakyrelu to leaky_relu (#9270)

2025-02-26 13:22:08 -05:00

olmoe.py

don't pass model in convert_from_huggingface and convert_from_gguf (#10094)

2025-04-28 20:11:19 -04:00

openelm.py

nn.RMSNorm (#5272)

2024-07-02 21:39:01 -04:00

qwq.py

don't pass model in convert_from_huggingface and convert_from_gguf (#10094)

2025-04-28 20:11:19 -04:00

sdv2.py

move benchmark stat tracking to influxdb (#10185)

2025-05-15 16:14:56 -07:00

sdxl_seed0.png

default threefry (#6116)

2024-09-25 17:45:13 +08:00

sdxl.py

move benchmark stat tracking to influxdb (#10185)

2025-05-15 16:14:56 -07: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_seed0.png

default threefry (#6116)

2024-09-25 17:45:13 +08:00

stable_diffusion.py

move benchmark stat tracking to influxdb (#10185)

2025-05-15 16:14:56 -07:00

stunning_mnist.py

make beautiful indexing use a Variable (#10063)

2025-04-27 08:22:38 -04: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

Fix yolo load failing silently (#10046)

2025-05-07 11:46:09 -07:00