Metadata-Version: 2.1
Name: atorch
Version: 0.1.11rc1
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
Requires-Dist: fairscale ==0.4.1
Requires-Dist: apex
Requires-Dist: flash-attn
Requires-Dist: deepspeed ==0.10.3
Requires-Dist: transformers >=4.31.0
Requires-Dist: networkx
Requires-Dist: pyomo ==6.7.0
Requires-Dist: pynvml ==11.4.1
Requires-Dist: fsspec
Requires-Dist: grpcio
Requires-Dist: grpcio-tools
Requires-Dist: pyarrow ==12.0.0
Requires-Dist: pandas ==2.0.1
Requires-Dist: tensorboard ==2.11.0
Requires-Dist: dlrover[torch] ==0.4.2
Requires-Dist: protobuf
Requires-Dist: safetensors
Requires-Dist: GPy
Requires-Dist: pymoo ==0.5.0
Requires-Dist: grouped-gemm
Requires-Dist: megablocks
Requires-Dist: aistudio-common >=0.0.28.31
Requires-Dist: oss2 ==2.17.0
Requires-Dist: easydl-sdk ==0.0.6
Requires-Dist: unifile-sdk ==0.1.14
Requires-Dist: alipay-pcache ==0.1.6
Requires-Dist: zdfs-dfs ==2.3.2

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.
