CITATION.cff
MANIFEST.in
README.md
pyproject.toml
nabla/__init__.py
nabla/config.py
nabla/testing.py
nabla/core/__init__.py
nabla/core/testing.py
nabla/core/autograd/__init__.py
nabla/core/autograd/backward.py
nabla/core/autograd/forward.py
nabla/core/common/__init__.py
nabla/core/common/context.py
nabla/core/common/pytree.py
nabla/core/graph/__init__.py
nabla/core/graph/engine.py
nabla/core/graph/tracing.py
nabla/core/graph/utils.py
nabla/core/sharding/__init__.py
nabla/core/sharding/propagation.py
nabla/core/sharding/spec.py
nabla/core/sharding/spmd.py
nabla/core/sharding/optimizer/auto_sharding.py
nabla/core/tensor/__init__.py
nabla/core/tensor/api.py
nabla/core/tensor/impl.py
nabla/nn/__init__.py
nabla/nn/finetune/__init__.py
nabla/nn/finetune/checkpoint.py
nabla/nn/finetune/lora.py
nabla/nn/finetune/qlora.py
nabla/nn/functional/__init__.py
nabla/nn/functional/init.py
nabla/nn/functional/layers.py
nabla/nn/functional/losses.py
nabla/nn/modules/__init__.py
nabla/nn/modules/activation.py
nabla/nn/modules/attention.py
nabla/nn/modules/base.py
nabla/nn/modules/containers.py
nabla/nn/modules/dropout.py
nabla/nn/modules/embedding.py
nabla/nn/modules/layernorm.py
nabla/nn/modules/linear.py
nabla/nn/modules/transformer.py
nabla/nn/optim/__init__.py
nabla/nn/optim/functional.py
nabla/nn/optim/optimizer.py
nabla/ops/__init__.py
nabla/ops/base.py
nabla/ops/binary.py
nabla/ops/comparison.py
nabla/ops/control_flow.py
nabla/ops/convolution.py
nabla/ops/creation.py
nabla/ops/multi_output.py
nabla/ops/pooling.py
nabla/ops/reduction.py
nabla/ops/unary.py
nabla/ops/utils.py
nabla/ops/communication/__init__.py
nabla/ops/communication/all_gather.py
nabla/ops/communication/all_reduce.py
nabla/ops/communication/all_to_all.py
nabla/ops/communication/axis_index.py
nabla/ops/communication/base.py
nabla/ops/communication/broadcast.py
nabla/ops/communication/p_permute.py
nabla/ops/communication/reduce_scatter.py
nabla/ops/communication/reshard.py
nabla/ops/communication/shard.py
nabla/ops/communication/transfer.py
nabla/ops/view/__init__.py
nabla/ops/view/axes.py
nabla/ops/view/batch.py
nabla/ops/view/complex.py
nabla/ops/view/indexing.py
nabla/ops/view/shape.py
nabla/transforms/__init__.py
nabla/transforms/compile.py
nabla/transforms/grad.py
nabla/transforms/jacfwd.py
nabla/transforms/jacrev.py
nabla/transforms/jvp.py
nabla/transforms/shard_map.py
nabla/transforms/utils.py
nabla/transforms/vjp.py
nabla/transforms/vmap.py
nabla_ml.egg-info/PKG-INFO
nabla_ml.egg-info/SOURCES.txt
nabla_ml.egg-info/dependency_links.txt
nabla_ml.egg-info/requires.txt
nabla_ml.egg-info/top_level.txt