v1.0.1

d99506f3 · chenzk · 61e92904 · d99506f3 · d99506f3 · d99506f3
Commit d99506f3 authored Dec 03, 2024 by chenzk
20 changed files
--- a/parallel/data_parallel/utils.py
+++ b/parallel/data_parallel/utils.py
--- a/parallel/parameters.py
+++ b/parallel/parameters.py
--- a/parallel/pipeline_parallel/README.md
+++ b/parallel/pipeline_parallel/README.md
--- a/src/nanotron/parallel/pipeline_parallel/__init__.py
+++ b/src/nanotron/parallel/pipeline_parallel/__init__.py
--- a/parallel/pipeline_parallel/block.py
+++ b/parallel/pipeline_parallel/block.py
--- a/parallel/pipeline_parallel/context_manager.py
+++ b/parallel/pipeline_parallel/context_manager.py
--- a/parallel/pipeline_parallel/engine.py
+++ b/parallel/pipeline_parallel/engine.py
--- a/parallel/pipeline_parallel/functional.py
+++ b/parallel/pipeline_parallel/functional.py
--- a/parallel/pipeline_parallel/p2p.py
+++ b/parallel/pipeline_parallel/p2p.py
--- a/parallel/pipeline_parallel/state.py
+++ b/parallel/pipeline_parallel/state.py
--- a/parallel/pipeline_parallel/tensor_pointer.py
+++ b/parallel/pipeline_parallel/tensor_pointer.py
--- a/parallel/pipeline_parallel/utils.py
+++ b/parallel/pipeline_parallel/utils.py
--- a/parallel/sharded_parameters.py
+++ b/parallel/sharded_parameters.py
--- a/src/nanotron/parallel/tensor_parallel/__init__.py
+++ b/src/nanotron/parallel/tensor_parallel/__init__.py
--- a/parallel/tensor_parallel/distributed_differentiable_primitives.py
+++ b/parallel/tensor_parallel/distributed_differentiable_primitives.py
--- a/parallel/tensor_parallel/enum.py
+++ b/parallel/tensor_parallel/enum.py
--- a/parallel/tensor_parallel/functional.py
+++ b/parallel/tensor_parallel/functional.py
--- a/parallel/tensor_parallel/nn.py
+++ b/parallel/tensor_parallel/nn.py
--- a/parallel/tied_parameters.py
+++ b/parallel/tied_parameters.py
--- a/parallel/utils.py
+++ b/parallel/utils.py