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

..

actions

split ocelot out for autogen, fix CI (#10819)

2025-06-15 11:37:23 -07:00

workflows

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

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