geohot 87affa8661 num_batches_tracked has shape ()

2025-10-21 09:10:35 +08:00

..

__init__.py

num_batches_tracked has shape ()

2025-10-21 09:10:35 +08:00

datasets.py

tar_extract with Tensors (#7853)

2024-12-04 17:03:19 +08:00

onnx.py

fix shape of m and v in onnx Adam with FUSE_OPTIM (#12768)

2025-10-17 10:32:41 -04:00

optim.py

remove a contiguous in LARS (#12770)

2025-10-17 17:07:30 -04:00

state.py

few shapetracker cleanups (#12741)

2025-10-16 12:43:27 -04:00

torch.py

move frontend dir to nn [pr] (#12470)

2025-10-07 10:42:22 +08:00