path # lines of code number of updates days since first update days since last update first updated last updated stubs/torch/nn/parallel/distributed.pyi 22 1 348 348 2021-02-12 2021-02-12 stubs/torch/nn/__init__.pyi 4 1 441 441 2020-11-11 2020-11-11 stubs/torch/nn/parameter.pyi 18 6 553 86 2020-07-22 2021-11-01 stubs/torch/nn/functional.pyi 177 3 553 82 2020-07-22 2021-11-05 stubs/torch/nn/modules/module.pyi 57 8 553 82 2020-07-22 2021-11-05 stubs/torch/nn/modules/__init__.pyi 48 2 553 330 2020-07-22 2021-03-02 stubs/torch/nn/modules/batchnorm.pyi 26 1 327 327 2021-03-05 2021-03-05 stubs/torch/nn/modules/linear.pyi 28 1 82 82 2021-11-05 2021-11-05 stubs/torch/nn/modules/adaptive.pyi 22 1 75 75 2021-11-12 2021-11-12 stubs/torch/nn/modules/activation.pyi 117 1 330 330 2021-03-02 2021-03-02 stubs/torch/jit.pyi 2 1 142 142 2021-09-06 2021-09-06 stubs/torch/__init__.pyi 1856 17 553 142 2020-07-22 2021-09-06 stubs/torch/distributed/distributed_c10d.pyi 4 2 441 79 2020-11-11 2021-11-08 stubs/torch/distributed/nn/functional.pyi 4 1 264 264 2021-05-07 2021-05-07 stubs/torch/distributed/__init__.pyi 68 12 553 79 2020-07-22 2021-11-08 stubs/torch/distributed/rpc/__init__.pyi 44 6 495 304 2020-09-18 2021-03-28 stubs/torch/autograd/__init__.pyi 37 2 553 463 2020-07-22 2020-10-20 stubs/torch/multiprocessing/__init__.pyi 10 2 379 364 2021-01-12 2021-01-27 stubs/torch/cuda/__init__.pyi 69 8 553 260 2020-07-22 2021-05-11 stubs/torch/cuda/comm/__init__.pyi 19 1 79 79 2021-11-08 2021-11-08 stubs/torch/cuda/amp/__init__.pyi 6 2 510 337 2020-09-03 2021-02-23 stubs/torch/cuda/amp/grad_scaler.pyi 12 4 510 250 2020-09-03 2021-05-21 stubs/torch/backends/cudnn.pyi 3 2 553 79 2020-07-22 2021-11-08 stubs/torch/futures.pyi 3 1 441 441 2020-11-11 2020-11-11 stubs/torch/optim/optimizer.pyi 13 4 516 328 2020-08-28 2021-03-04 stubs/torch/utils/__init__.pyi 1 2 553 545 2020-07-22 2020-07-30 stubs/torch/utils/checkpoint.pyi 7 4 553 302 2020-07-22 2021-03-30 stubs/torch/utils/data/distributed.pyi 8 1 497 497 2020-09-16 2020-09-16 stubs/torch/utils/data/dataset.pyi 22 1 497 497 2020-09-16 2020-09-16 stubs/torch/testing/__init__.pyi 2 1 370 370 2021-01-21 2021-01-21 stubs/torch/functional.pyi 5 3 553 436 2020-07-22 2020-11-16 stubs/torch/serialization.pyi 7 2 530 21 2020-08-14 2022-01-05 codecov.yml 19 2 253 169 2021-05-18 2021-08-10 fairscale/experimental/tooling/layer_memory_tracker.py 465 2 98 75 2021-10-20 2021-11-12 fairscale/experimental/tooling/__init__.py 1 1 98 98 2021-10-20 2021-10-20 fairscale/experimental/nn/ampnet_pipe/ampnet.py 303 4 337 75 2021-02-23 2021-11-12 fairscale/experimental/nn/ampnet_pipe/__init__.py 2 2 337 306 2021-02-23 2021-03-26 fairscale/experimental/nn/ampnet_pipe/pipe.py 47 3 337 75 2021-02-23 2021-11-12 fairscale/experimental/nn/auto_shard.py 121 4 229 91 2021-06-11 2021-10-27 fairscale/experimental/nn/data_parallel/gossip/mixing_manager.py 29 2 79 75 2021-11-08 2021-11-12 fairscale/experimental/nn/data_parallel/gossip/__init__.py 14 1 79 79 2021-11-08 2021-11-08 fairscale/experimental/nn/data_parallel/gossip/utils/__init__.py 9 1 79 79 2021-11-08 2021-11-08 fairscale/experimental/nn/data_parallel/gossip/utils/cuda_metering.py 57 2 79 75 2021-11-08 2021-11-12 fairscale/experimental/nn/data_parallel/gossip/utils/helpers.py 70 2 79 75 2021-11-08 2021-11-12 fairscale/experimental/nn/data_parallel/gossip/distributed.py 758 2 79 75 2021-11-08 2021-11-12 fairscale/experimental/nn/data_parallel/gossip/graph_manager.py 197 2 79 75 2021-11-08 2021-11-12 fairscale/experimental/nn/data_parallel/gossip/gossiper.py 181 2 79 75 2021-11-08 2021-11-12 fairscale/experimental/nn/data_parallel/__init__.py 1 1 79 79 2021-11-08 2021-11-08 fairscale/experimental/nn/distributed_pipeline/trace.py 80 1 273 273 2021-04-28 2021-04-28 fairscale/experimental/nn/distributed_pipeline/graph.py 114 4 288 75 2021-04-13 2021-11-12 fairscale/experimental/nn/distributed_pipeline/__init__.py 3 1 288 288 2021-04-13 2021-04-13 fairscale/experimental/nn/distributed_pipeline/partition_handler.py 178 4 288 75 2021-04-13 2021-11-12 fairscale/experimental/nn/distributed_pipeline/loss.py 10 1 288 288 2021-04-13 2021-04-13 fairscale/experimental/nn/distributed_pipeline/pipeline.py 117 4 288 72 2021-04-13 2021-11-15 fairscale/experimental/nn/distributed_pipeline/data.py 8 1 288 288 2021-04-13 2021-04-13 fairscale/experimental/nn/__init__.py 6 6 337 82 2021-02-23 2021-11-05 fairscale/experimental/nn/sync_batchnorm.py 133 6 264 142 2021-05-07 2021-09-06 fairscale/experimental/nn/mevo.py 317 4 82 44 2021-11-05 2021-12-13 fairscale/experimental/nn/ssd_offload.py 197 6 86 6 2021-11-01 2022-01-20 fairscale/experimental/nn/offload.py 281 10 334 75 2021-02-26 2021-11-12 fairscale/experimental/optim/dynamic_loss_scaler.py 124 2 267 75 2021-05-04 2021-11-12 fairscale/experimental/optim/__init__.py 3 1 267 267 2021-05-04 2021-05-04 fairscale/experimental/__init__.py 3 3 337 267 2021-02-23 2021-05-04 fairscale/version.py 1 2 36 12 2021-12-21 2022-01-14 fairscale/nn/misc/flatten_params_wrapper.py 303 16 370 8 2021-01-21 2022-01-18 fairscale/nn/misc/__init__.py 5 6 370 259 2021-01-21 2021-05-12 fairscale/nn/misc/param_bucket.py 145 4 313 75 2021-03-19 2021-11-12 fairscale/nn/data_parallel/fully_sharded_data_parallel.py 1260 67 337 6 2021-02-23 2022-01-20 fairscale/nn/data_parallel/__init__.py 4 7 538 70 2020-08-06 2021-11-17 fairscale/nn/data_parallel/fsdp_optim_utils.py 137 6 307 69 2021-03-25 2021-11-18 fairscale/nn/data_parallel/sharded_ddp.py 352 43 495 41 2020-09-18 2021-12-16 fairscale/nn/pipe/worker.py 101 2 495 75 2020-09-18 2021-11-12 fairscale/nn/pipe/phony.py 15 1 288 288 2021-04-13 2021-04-13 fairscale/nn/pipe/checkpoint.py 165 2 495 75 2020-09-18 2021-11-12 fairscale/nn/pipe/__init__.py 6 5 441 169 2020-11-11 2021-08-10 fairscale/nn/pipe/balance/profile.py 77 2 495 75 2020-09-18 2021-11-12 fairscale/nn/pipe/balance/__init__.py 34 1 411 411 2020-12-11 2020-12-11 fairscale/nn/pipe/copy.py 68 1 75 75 2021-11-12 2021-11-12 fairscale/nn/pipe/types.py 52 6 441 360 2020-11-11 2021-01-31 fairscale/nn/pipe/stream.py 55 1 495 495 2020-09-18 2020-09-18 fairscale/nn/pipe/skip/portal.py 119 2 495 75 2020-09-18 2021-11-12 fairscale/nn/pipe/skip/layout.py 48 2 495 75 2020-09-18 2021-11-12 fairscale/nn/pipe/skip/tracker.py 98 2 495 75 2020-09-18 2021-11-12 fairscale/nn/pipe/skip/skippable.py 179 2 495 75 2020-09-18 2021-11-12 fairscale/nn/pipe/rpc.py 200 4 441 352 2020-11-11 2021-02-08 fairscale/nn/pipe/async_schedule.py 364 8 441 75 2020-11-11 2021-11-12 fairscale/nn/pipe/pipeline.py 143 5 495 75 2020-09-18 2021-11-12 fairscale/nn/pipe/microbatch.py 107 1 495 495 2020-09-18 2020-09-18 fairscale/nn/pipe/messages.py 115 2 441 379 2020-11-11 2021-01-12 fairscale/nn/pipe/async_pipe.py 225 8 362 75 2021-01-29 2021-11-12 fairscale/nn/pipe/pipe.py 182 9 495 75 2020-09-18 2021-11-12 fairscale/nn/pipe/async_pipeline.py 65 3 360 75 2021-01-31 2021-11-12 fairscale/nn/checkpoint/checkpoint_activations.py 167 9 259 75 2021-05-12 2021-11-12 fairscale/nn/checkpoint/__init__.py 3 2 259 244 2021-05-12 2021-05-27 fairscale/nn/checkpoint/checkpoint_utils.py 21 3 259 136 2021-05-12 2021-09-12 fairscale/nn/__init__.py 8 12 481 253 2020-10-02 2021-05-18 fairscale/nn/model_parallel/cross_entropy.py 48 2 553 218 2020-07-22 2021-06-22 fairscale/nn/model_parallel/initialize.py 92 5 553 332 2020-07-22 2021-02-28 fairscale/nn/model_parallel/__init__.py 19 3 553 306 2020-07-22 2021-03-26 fairscale/nn/model_parallel/random.py 114 2 553 441 2020-07-22 2020-11-11 fairscale/nn/model_parallel/utils.py 28 1 553 553 2020-07-22 2020-07-22 fairscale/nn/model_parallel/layers.py 225 3 553 495 2020-07-22 2020-09-18 fairscale/nn/model_parallel/mappings.py 69 3 553 441 2020-07-22 2020-11-11 fairscale/nn/wrap/auto_wrap.py 108 5 330 75 2021-03-02 2021-11-12 fairscale/nn/wrap/__init__.py 3 4 330 253 2021-03-02 2021-05-18 fairscale/nn/moe/__init__.py 4 4 481 306 2020-10-02 2021-03-26 fairscale/nn/moe/top2gate.py 67 7 481 75 2020-10-02 2021-11-12 fairscale/nn/moe/moe_layer.py 53 6 466 245 2020-10-17 2021-05-26 fairscale/clib/fused_adam_cuda/fused_adam_cuda.cpp 5 3 544 522 2020-07-31 2020-08-22 fairscale/clib/fused_adam_cuda/fused_adam_cuda_kernel.cu 246 6 544 98 2020-07-31 2021-10-20 fairscale/clib/fused_adam_cuda/compat.h 4 1 544 544 2020-07-31 2020-07-31 fairscale/clib/fused_adam_cuda/multi_tensor_apply.cuh 111 1 544 544 2020-07-31 2020-07-31 fairscale/optim/grad_scaler.py 247 10 510 13 2020-09-03 2022-01-13 fairscale/optim/__init__.py 13 8 510 306 2020-09-03 2021-03-26 fairscale/optim/oss.py 339 67 544 75 2020-07-31 2021-11-12 fairscale/optim/adam.py 192 7 544 522 2020-07-31 2020-08-22 fairscale/optim/layerwise_gradient_scaler.py 145 2 13 12 2022-01-13 2022-01-14 fairscale/optim/adascale.py 226 20 467 75 2020-10-16 2021-11-12 fairscale/__init__.py 5 28 544 36 2020-07-31 2021-12-21 fairscale/utils/reduce_scatter_bucketer.py 106 6 337 75 2021-02-23 2021-11-12 fairscale/utils/params.py 40 2 263 142 2021-05-08 2021-09-06 fairscale/utils/state_dict.py 24 1 334 334 2021-02-26 2021-02-26 fairscale/utils/version.py 12 1 214 214 2021-06-26 2021-06-26 fairscale/utils/testing_memory.py 14 2 363 75 2021-01-28 2021-11-12 fairscale/utils/object.py 18 1 441 441 2020-11-11 2020-11-11 fairscale/utils/containers.py 77 4 350 51 2021-02-10 2021-12-06 fairscale/utils/__init__.py 1 3 432 214 2020-11-20 2021-06-26 fairscale/utils/parallel.py 55 8 337 19 2021-02-23 2022-01-07 fairscale/utils/testing.py 495 36 420 75 2020-12-02 2021-11-12 fairscale/utils/golden_testing_data.py 8 2 363 338 2021-01-28 2021-02-22 pyproject.toml 28 22 543 13 2020-08-01 2022-01-13 MANIFEST.in 3 2 543 420 2020-08-01 2020-12-02 setup.py 59 15 544 36 2020-07-31 2021-12-21 benchmarks/experimental/experimental_async_approaches.py 597 6 300 69 2021-04-01 2021-11-18 benchmarks/experimental/benchmark_mevo.py 107 2 82 75 2021-11-05 2021-11-12 benchmarks/experimental/benchmark_dataset.py 40 3 322 75 2021-03-10 2021-11-12 benchmarks/experimental/sync_batchnorm.py 28 1 264 264 2021-05-07 2021-05-07 benchmarks/experimental/offload.py 328 8 334 75 2021-02-26 2021-11-12 benchmarks/golden_configs/lm_wikitext2.py 103 11 372 63 2021-01-19 2021-11-24 benchmarks/golden_configs/oss_mnist.py 8 7 366 95 2021-01-25 2021-10-23 benchmarks/golden_configs/__init__.py 1 1 372 372 2021-01-19 2021-01-19 benchmarks/datasets/__init__.py 1 1 372 372 2021-01-19 2021-01-19 benchmarks/datasets/wikitext2_data.py 73 11 372 63 2021-01-19 2021-11-24 benchmarks/datasets/mnist.py 23 3 328 263 2021-03-04 2021-05-08 benchmarks/__init__.py 1 1 320 320 2021-03-12 2021-03-12 benchmarks/models/__init__.py 1 1 372 372 2021-01-19 2021-01-19 benchmarks/models/transformer_lm.py 115 3 372 63 2021-01-19 2021-11-24 benchmarks/oss.py 254 32 522 75 2020-08-22 2021-11-12 benchmarks/utils.py 124 1 63 63 2021-11-24 2021-11-24 benchmarks/moe.py 98 1 63 63 2021-11-24 2021-11-24 benchmarks/pipe.py 235 23 510 63 2020-09-03 2021-11-24 setup.cfg 34 6 558 214 2020-07-17 2021-06-26 release_utils.py 49 1 36 36 2021-12-21 2021-12-21