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_multigpu.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

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_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

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

llama3.py

update llama3 70B to use R1 (#8733)

2025-01-23 19:06:05 -05:00

llama.py

validate llama quantize output (#7901)

2024-11-25 16:46:23 -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_seed0.png

default threefry (#6116)

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

sdxl.py

GlobalCounters.reset() in sdxl step [pr] (#8664)

2025-01-17 21:10:28 -05: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_seed0.png

default threefry (#6116)

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

stable_diffusion.py

Remove wgpu specific checks from stable diffusion example (#7991)

2024-12-02 11:31:14 +01: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