Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/axonn-ai/axonn
A parallel framework for training deep neural networks
https://github.com/axonn-ai/axonn
deep-learning machine-learning neural-networks parallel-computing
Last synced: 11 days ago
JSON representation
A parallel framework for training deep neural networks
- Host: GitHub
- URL: https://github.com/axonn-ai/axonn
- Owner: axonn-ai
- License: apache-2.0
- Created: 2021-06-01T20:04:27.000Z (over 3 years ago)
- Default Branch: develop
- Last Pushed: 2024-04-14T02:21:20.000Z (7 months ago)
- Last Synced: 2024-04-14T11:19:56.158Z (7 months ago)
- Topics: deep-learning, machine-learning, neural-networks, parallel-computing
- Language: Python
- Homepage: https://axonn.readthedocs.io
- Size: 24.5 MB
- Stars: 29
- Watchers: 4
- Forks: 4
- Open Issues: 15
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# AxoNN
[![nvidia-rtx-3090-tests](https://github.com/axonn-ai/axonn/actions/workflows/ci.yaml/badge.svg)](https://github.com/axonn-ai/axonn/actions/workflows/ci.yaml)
[![docs](https://readthedocs.org/projects/axonn/badge/?version=latest)](https://axonn.readthedocs.io/en/latest/?badge=latest)
[![Code Style: Black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
[![Join slack](https://img.shields.io/badge/slack-axonn--users-blue)](https://join.slack.com/t/axonn-users/shared_invite/zt-2itbahk29-_Ig1JasFxnuVyfMtcC4GnA)AxoNN is a parallel framework for training deep neural networks.
### Installation
Prior to the installation, [PyTorch](https://pytorch.org/get-started/locally/) must already be installed.```bash
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](https://pssg.cs.umd.edu/assets/papers/2022-05-axonn-ipdps.pdf):* 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](https://github.com/pssg-int/axonn/blob/develop/LICENSE) for
details.SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception