LICENSE
README.md
pyproject.toml
axonn/__init__.py
axonn/axonn.py
axonn/checkpoint.py
axonn/communication.py
axonn/config.py
axonn/inter_layer.py
axonn/optim.py
axonn/utils.py
axonn.egg-info/PKG-INFO
axonn.egg-info/SOURCES.txt
axonn.egg-info/dependency_links.txt
axonn.egg-info/requires.txt
axonn.egg-info/top_level.txt
axonn/intra_layer/__init__.py
axonn/intra_layer/asym_communication.py
axonn/intra_layer/automatic_parallelism.py
axonn/intra_layer/communication.py
axonn/intra_layer/conv.py
axonn/intra_layer/embedding.py
axonn/intra_layer/fully_connected.py
axonn/intra_layer/gradient_normalization.py
axonn/intra_layer/overlap_communication.py
axonn/intra_layer/utils.py
axonn/lightning/__init__.py
axonn/lightning/axonn_strategy.py
axonn/models/__init__.py
axonn/models/transformers/__init__.py
axonn/models/transformers/modify_llama.py
axonn/models/transformers/modify_mistral.py
axonn/models/transformers/modify_mixtral.py
axonn/models/transformers/modify_opt.py