Alexey Zaytsev 230ad3a460 [bounty] Don't use numpy inside hlb_cifar10 training loop (#10777) ...

* Don't use numpy inside hlb_cifar10 training loop

* Lint it

* jit it

* Drop the last half-batch

* Use gather for random_crop and reuse perms

* Wrap train_cifar in FUSE_ARANGE context

* No need to pass FUSE_ARANGE=1 to hlb_cifar10.py

* Add cutmix to jittable augmentations

* Remove .contiguous() from fetch_batches

* Fix indexing boundary

---------

Co-authored-by: Irwin1138 <irwin1139@gmail.com>

2025-06-23 17:24:56 -07:00

..

benchmark_search.yml

[bounty] Don't use numpy inside hlb_cifar10 training loop (#10777)

2025-06-23 17:24:56 -07:00

benchmark.yml

remove comma 0.9.4 from benchmark (#10867)

2025-06-18 12:43:59 -04:00

docs.yml

add strict mkdocs check (#5497)

2024-07-15 14:21:37 -07:00

mlperf.yml

remove AMD_LLVM=0 in mlperf and search ci (#10785)

2025-06-11 21:10:31 -04:00

python-publish.yml

update gh actions (#3033)

2024-01-09 17:52:22 -08:00

szdiff.yml

update gh actions (#3033)

2024-01-09 17:52:22 -08:00

test.yml

Revert "change clang -march flag to -mcpu on arm (#10841)" (#10942)

2025-06-23 16:48:28 -07:00