chenyu ff05bff221 put bert data shard inside jit (#9160) ... python time 45ms -> 9ms, it was spending time to schedule the shard also init bert data on CLANG since it's from numpy, so we don't create the tensor on default device then shard into GPUS		2025-02-18 10:36:54 -05:00
..
conversation_data	Whisper + LLAMA + VITS (#2332)	2023-12-02 15:03:46 -08:00
llm.c	create_schedule([x.lazydata]) -> x.schedule() in tests (#8449)	2024-12-31 03:15:52 +08:00
mlperf	put bert data shard inside jit (#9160)	2025-02-18 10:36:54 -05:00
openpilot	add compile3 benchmark [pr] (#8929)	2025-02-06 22:49:31 +08:00
other_mnist	beautiful_mnist in torch	2024-07-14 11:09:58 -07: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	hotfix: add missing tinychat fonts + other assets	2025-02-01 09:34:44 +08:00
vgg7_helpers	move to new cached fetch (#2493)	2023-11-28 17:36:55 -08:00
webgpu	WebGPU f16 support (f16 bounty part 2) (#8653)	2025-02-12 19:46:53 +08: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	add onnx_helpers to extra and add ort validate to benchmark_onnx (#8890)	2025-02-04 16:36:01 -05:00
coder.py	apply the same fix_bf16 in llama and coder (#3789)	2024-03-17 21:25:24 -04:00
compile_efficientnet.py	Remove WebGL (#8012)	2024-12-03 16:02:53 +01:00
compile_tensorflow.py	make onnx runner a class (#8647)	2025-01-20 10:11:05 -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	cleanup ci, split docs/autogen, testing_minimal, LLVM Speed [pr] (#8952)	2025-02-07 19:01:59 +08:00
handcode_opt.py	move time_linearizer to extra.optimization.helpers [pr] (#9048)	2025-02-12 15:49:58 -05: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	update llama3 70B to use R1 (#8733)	2025-01-23 19:06:05 -05: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	tinytqdm.set_description and tinytrange (#5101)	2024-06-22 14:45:06 -04: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	combine pad2d with pad (#7677)	2024-11-14 17:56:02 +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	hotfix: add replay_pkl debugging env	2025-02-17 17:34:58 +08: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	fix onehot and jit in examples/transformer (#3073)	2024-01-10 02:22:41 -05: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	Fix examples/conversation.py (#8425)	2024-12-26 12:45:19 -05:00
whisper.py	enable whisper batch for long sequences (#6458)	2024-09-17 00:42:10 -04:00
yolov3.py	Update yolov3.py (#2680)	2023-12-08 12:59:38 -08:00
yolov8-onnx.py	add onnx_helpers to extra and add ort validate to benchmark_onnx (#8890)	2025-02-04 16:36:01 -05:00
yolov8.py	YoloV8 on WebGPU (#8007)	2024-12-03 15:10:41 +01:00