LICENSE
MANIFEST.in
README.md
pyproject.toml
requirements.txt
setup.cfg
setup.py
fairscale/__init__.py
fairscale.egg-info/PKG-INFO
fairscale.egg-info/SOURCES.txt
fairscale.egg-info/dependency_links.txt
fairscale.egg-info/requires.txt
fairscale.egg-info/top_level.txt
fairscale/clib/fused_adam_cuda/fused_adam_cuda.cpp
fairscale/clib/fused_adam_cuda/fused_adam_cuda_kernel.cu
fairscale/nn/__init__.py
fairscale/nn/data_parallel/__init__.py
fairscale/nn/data_parallel/sharded_ddp.py
fairscale/nn/model_parallel/__init__.py
fairscale/nn/model_parallel/cross_entropy.py
fairscale/nn/model_parallel/initialize.py
fairscale/nn/model_parallel/layers.py
fairscale/nn/model_parallel/mappings.py
fairscale/nn/model_parallel/random.py
fairscale/nn/model_parallel/utils.py
fairscale/nn/moe/__init__.py
fairscale/nn/moe/moelayer.py
fairscale/nn/moe/top2gate.py
fairscale/nn/pipe/__init__.py
fairscale/nn/pipe/batchnorm.py
fairscale/nn/pipe/checkpoint.py
fairscale/nn/pipe/copy.py
fairscale/nn/pipe/dependency.py
fairscale/nn/pipe/microbatch.py
fairscale/nn/pipe/phony.py
fairscale/nn/pipe/pipe.py
fairscale/nn/pipe/pipeline.py
fairscale/nn/pipe/stream.py
fairscale/nn/pipe/worker.py
fairscale/nn/pipe/balance/__init__.py
fairscale/nn/pipe/balance/blockpartition.py
fairscale/nn/pipe/balance/profile.py
fairscale/nn/pipe/skip/__init__.py
fairscale/nn/pipe/skip/layout.py
fairscale/nn/pipe/skip/namespace.py
fairscale/nn/pipe/skip/portal.py
fairscale/nn/pipe/skip/skippable.py
fairscale/nn/pipe/skip/tracker.py
fairscale/optim/__init__.py
fairscale/optim/adam.py
fairscale/optim/grad_scaler.py
fairscale/optim/oss.py
fairscale/optim/utils.py