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.yml	remove comma 0.9.4 from benchmark (#10867)	2025-06-18 12:43:59 -04:00
benchmark_search.yml	[bounty] Don't use numpy inside hlb_cifar10 training loop (#10777)	2025-06-23 17:24:56 -07: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