Name | Modified | Size | Downloads / Week |
---|---|---|---|
Parent folder | |||
README.md | 2024-04-22 | 551 Bytes | |
v0.7.0 source code.tar.gz | 2024-04-22 | 1.5 MB | |
v0.7.0 source code.zip | 2024-04-22 | 1.8 MB | |
Totals: 3 Items | 3.3 MB | 0 |
No major features in this release
Changelog
- Expanding out ZCH/MCH
- Increased support with Torch Dynamo/Export
- Distributed Benchmarking introduced under torchrec/distributed/benchmarks for inference and training
- VBE optimizations
- TWRW support for VBE (I think this happened in the last release, Josh can confirm)
- Generalized train_pipeline for different pipeline stage overlapping
- Autograd support for traceable collectives
- Output dtype support for embeddings
- Dynamo tracing for sharded embedding modules
- Bug fixes