George HotzandGitHub 9d72119a0c minor resnet cleanups (#6382) ...

* minor resnet cleanups

* that should have been long

* jit

* meh

2024-09-06 12:50:21 +08:00

..

__init__.py

minor resnet cleanups (#6382)

2024-09-06 12:50:21 +08:00

datasets.py

add cifar to datasets.py (#6210)

2024-08-20 11:42:49 -07:00

optim.py

nn docs (#4735)

2024-05-26 13:08:22 -07:00

state.py

faster local tests. copy torch permuted to defautl device [run_process_replay] (#6363)

2024-09-05 13:57:20 +08:00