LICENSE
MANIFEST.in
NOTICE
README.md
pyproject.toml
requirements.txt
setup.cfg
setup.py
fairscale/__init__.py
fairscale/version.py
fairscale.egg-info/PKG-INFO
fairscale.egg-info/SOURCES.txt
fairscale.egg-info/dependency_links.txt
fairscale.egg-info/entry_points.txt
fairscale.egg-info/requires.txt
fairscale.egg-info/top_level.txt
fairscale/clib/fused_adam_cuda/compat.h
fairscale/clib/fused_adam_cuda/multi_tensor_apply.cuh
fairscale/experimental/__init__.py
fairscale/experimental/nn/__init__.py
fairscale/experimental/nn/auto_shard.py
fairscale/experimental/nn/mevo.py
fairscale/experimental/nn/offload.py
fairscale/experimental/nn/sync_batchnorm.py
fairscale/experimental/nn/ampnet_pipe/__init__.py
fairscale/experimental/nn/ampnet_pipe/ampnet.py
fairscale/experimental/nn/ampnet_pipe/pipe.py
fairscale/experimental/nn/data_parallel/__init__.py
fairscale/experimental/nn/data_parallel/gossip/__init__.py
fairscale/experimental/nn/data_parallel/gossip/distributed.py
fairscale/experimental/nn/data_parallel/gossip/gossiper.py
fairscale/experimental/nn/data_parallel/gossip/graph_manager.py
fairscale/experimental/nn/data_parallel/gossip/mixing_manager.py
fairscale/experimental/nn/data_parallel/gossip/utils/__init__.py
fairscale/experimental/nn/data_parallel/gossip/utils/cuda_metering.py
fairscale/experimental/nn/data_parallel/gossip/utils/helpers.py
fairscale/experimental/nn/distributed_pipeline/__init__.py
fairscale/experimental/nn/distributed_pipeline/data.py
fairscale/experimental/nn/distributed_pipeline/graph.py
fairscale/experimental/nn/distributed_pipeline/loss.py
fairscale/experimental/nn/distributed_pipeline/partition_handler.py
fairscale/experimental/nn/distributed_pipeline/pipeline.py
fairscale/experimental/nn/distributed_pipeline/trace.py
fairscale/experimental/optim/__init__.py
fairscale/experimental/optim/dynamic_loss_scaler.py
fairscale/experimental/tooling/__init__.py
fairscale/experimental/tooling/layer_memory_tracker.py
fairscale/experimental/wgit/__init__.py
fairscale/experimental/wgit/__main__.py
fairscale/experimental/wgit/cli.py
fairscale/experimental/wgit/pygit.py
fairscale/experimental/wgit/repo.py
fairscale/experimental/wgit/sha1_store.py
fairscale/experimental/wgit/signal_sparsity.py
fairscale/experimental/wgit/signal_sparsity_profiling.py
fairscale/experimental/wgit/utils.py
fairscale/experimental/wgit/version.py
fairscale/fair_dev/__init__.py
fairscale/fair_dev/common_paths.py
fairscale/fair_dev/testing/__init__.py
fairscale/fair_dev/testing/golden_testing_data.py
fairscale/fair_dev/testing/testing.py
fairscale/fair_dev/testing/testing_memory.py
fairscale/internal/__init__.py
fairscale/internal/containers.py
fairscale/internal/object.py
fairscale/internal/parallel.py
fairscale/internal/params.py
fairscale/internal/reduce_scatter_bucketer.py
fairscale/internal/state_dict.py
fairscale/internal/version.py
fairscale/nn/__init__.py
fairscale/nn/checkpoint/__init__.py
fairscale/nn/checkpoint/checkpoint_activations.py
fairscale/nn/checkpoint/checkpoint_utils.py
fairscale/nn/data_parallel/__init__.py
fairscale/nn/data_parallel/fsdp_optim_utils.py
fairscale/nn/data_parallel/fully_sharded_data_parallel.py
fairscale/nn/data_parallel/sharded_ddp.py
fairscale/nn/misc/__init__.py
fairscale/nn/misc/flatten_params_wrapper.py
fairscale/nn/misc/param_bucket.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/moe_layer.py
fairscale/nn/moe/top2gate.py
fairscale/nn/pipe/__init__.py
fairscale/nn/pipe/async_pipe.py
fairscale/nn/pipe/async_pipeline.py
fairscale/nn/pipe/async_schedule.py
fairscale/nn/pipe/batchnorm.py
fairscale/nn/pipe/checkpoint.py
fairscale/nn/pipe/copy.py
fairscale/nn/pipe/dependency.py
fairscale/nn/pipe/messages.py
fairscale/nn/pipe/microbatch.py
fairscale/nn/pipe/phony.py
fairscale/nn/pipe/pipe.py
fairscale/nn/pipe/pipeline.py
fairscale/nn/pipe/py.typed
fairscale/nn/pipe/rpc.py
fairscale/nn/pipe/stream.py
fairscale/nn/pipe/types.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/balance/py.typed
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/nn/wrap/__init__.py
fairscale/nn/wrap/auto_wrap.py
fairscale/optim/__init__.py
fairscale/optim/adam.py
fairscale/optim/adascale.py
fairscale/optim/grad_scaler.py
fairscale/optim/layerwise_gradient_scaler.py
fairscale/optim/oss.py