Metadata-Version: 2.1
Name: atorch
Version: 0.2.4
Summary: A pytorch extension for efficient deep learning.
Home-page: https://github.com/intelligent-machine-learning/dlrover/tree/master/atorch
Author: Ant Group
Requires-Python: >=3.8
License-File: LICENSE
Requires-Dist: transformers >=4.31.0
Requires-Dist: networkx
Requires-Dist: pynvml >=11.4.1
Requires-Dist: grpcio
Requires-Dist: pyarrow >=12.0.0
Requires-Dist: pandas >=2.0.1
Requires-Dist: tensorboard >=2.11.0
Requires-Dist: dlrover[torch]
Requires-Dist: protobuf
Requires-Dist: safetensors

ATorch supports efficient and easy-to-use model training experience. ATorch provides performance optimizations in aspects such as I/O, preprocessing, computation, and communication (including automatic optimization), and has supported large-scale pretraining and finetuning of LLMs with over 100 billion parameters and thousands of advanced GPUs.
