Skip to main content

axonn AxoNN

nvidia-rtx-3090-tests

docs Code Style: Black Join slack

AxoNN is a parallel framework for training deep neural networks.

Installation

Prior to the installation, PyTorch must already be installed.

pip install axonn

Contributing

AxoNN is an open source project. We welcome contributions via pull requests, and questions, feature requests, or bug reports via issues.

Citing AxoNN

If you are referencing AxoNN in a publication, please cite the following paper:

  • Siddharth Singh, Abhinav Bhatele. AxoNN: An asynchronous, message-driven parallel framework for extreme-scale deep learning In Proceedings of the IEEE International Parallel & Distributed Processing Symposium (IPDPS '22). IEEE Computer Society, May 2022.

License

AxoNN is distributed under the terms of the Apache License (Version 2.0) with LLVM Exceptions.

All contributions must be made under the Apache License (Version 2.0) with LLVM Exceptions.

See LICENSE for details.

SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

Metadata

Release files for axonn 0.2.0

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for axonn 0.2.0
File Size Uploaded
axonn-0.2.0.tar.gz 37.7 kB Details

Built distribution (wheel)

Table of built distributions (wheels) for axonn 0.2.0
File Interpreter ABI Platform
axonn-0.2.0-py3-none-any.whl Python 3 none any Details

Total release size: 86.8 kB

Release files / axonn-0.2.0.tar.gz

Download URL axonn-0.2.0.tar.gz
Size 37.7 kB
Tags Source
SHA-256 checksum
How to use checksums
43fab74aeb5b0e49f81ff381d8302b071460f5a4bf6783cedd6136496f85ad57
BLAKE2b-256 checksum
How to use checksums
64d484f86e193883c8455a0cbaf836a31c5ae8fbe30c73367cd4da8b7400c387
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/5.1.1 CPython/3.11.2

Release files / axonn-0.2.0-py3-none-any.whl

Download URL axonn-0.2.0-py3-none-any.whl
Size 49.1 kB
Tags Python 3
SHA-256 checksum
How to use checksums
6f49d6713b79abb08196a26c195bc1968819b2aa07084a752e55c40912bdcb35
BLAKE2b-256 checksum
How to use checksums
530a1ebcfb49703879b8a745b412413507b19a301549dd50f9356e52db865e5a
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/5.1.1 CPython/3.11.2

Release history Release notifications | RSS feed

This release

0.2.0 This release

2 release files

0.1.0

1 release file

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page