A pytorch extension for efficient deep learning.
Project description
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.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
No source distribution files available for this release.See tutorial on generating distribution archives.
Built Distribution
atorch-0.2.1-py3-none-any.whl
(754.5 kB
view hashes)