Yao Matrix c8c1e525ed from 1.11.0, torchao.prototype.low_bit_optim is promoted to torchao.optim (#38689) ...

* since 1.11.0, torchao.prototype.low_bit_optim is promoted to
torchao.optim

Signed-off-by: YAO Matrix <matrix.yao@intel.com>

* fix review comments

Signed-off-by: YAO Matrix <matrix.yao@intel.com>

---------

Signed-off-by: YAO Matrix <matrix.yao@intel.com>
Co-authored-by: Marc Sun <57196510+SunMarc@users.noreply.github.com>

2025-06-11 12:16:25 +00:00

..

__init__.py

[Test refactor 1/5] Per-folder tests reorganization (#15725)

2022-02-23 15:46:28 -05:00

test_data_collator.py

add FlashAttentionKwargs and seq_idx to flat collator (#36456)

2025-04-16 15:45:03 +02:00

test_trainer_callback.py

fix: prevent second save in the end of training if last step was saved already (#36219)

2025-02-20 17:38:52 +01:00

test_trainer_distributed_loss.py

fix bug in distributed loss test (#38166)

2025-05-16 16:21:35 +00:00

test_trainer_distributed_worker_seed.py

enable trainer test cases on xpu (#38138)

2025-05-15 12:17:44 +00:00

test_trainer_distributed.py

Use Python 3.9 syntax in tests (#37343)

2025-04-08 14:12:08 +02:00

test_trainer_fsdp.py

fix FSDP + torch.compile bug when saving pretrained model (#37725)

2025-05-06 17:51:28 +02:00

test_trainer_seq2seq.py

[generation] Less verbose warnings by default (#38179)

2025-05-19 10:03:37 +00:00

test_trainer_tpu.py

Use Python 3.9 syntax in tests (#37343)

2025-04-08 14:12:08 +02:00

test_trainer_utils.py

Use Python 3.9 syntax in tests (#37343)

2025-04-08 14:12:08 +02:00

test_trainer.py

from 1.11.0, torchao.prototype.low_bit_optim is promoted to torchao.optim (#38689)

2025-06-11 12:16:25 +00:00