CITATION.cff
MANIFEST.in
README.md
pyproject.toml
readme.md
docs/conf.py
docs/conf_new.py
docs/conf_old.py
examples/add_one_custom.py
examples/conds_jit.py
examples/jvp_examples.py
examples/mlp_inference.py
examples/mlp_inference_jit.py
examples/mlp_train.py
examples/mlp_train_jax.py
examples/mlp_train_jit.py
examples/plot_basic_operations.py
examples/vjp_examples.py
examples/vmap_examples.py
nabla/__init__.py
nabla/core/__init__.py
nabla/core/array.py
nabla/core/execution_context.py
nabla/core/graph_execution.py
nabla/core/trafos.py
nabla/nn/__init__.py
nabla/nn/base.py
nabla/nn/activations/__init__.py
nabla/nn/activations/advanced.py
nabla/nn/activations/basic.py
nabla/nn/activations/parametric.py
nabla/nn/architectures/__init__.py
nabla/nn/architectures/mlp.py
nabla/nn/architectures/resnet.py
nabla/nn/architectures/rnn.py
nabla/nn/architectures/transformer.py
nabla/nn/init/__init__.py
nabla/nn/init/constant.py
nabla/nn/init/normal.py
nabla/nn/init/uniform.py
nabla/nn/init/variance_scaling.py
nabla/nn/layers/__init__.py
nabla/nn/layers/attention.py
nabla/nn/layers/conv.py
nabla/nn/layers/embedding.py
nabla/nn/layers/linear.py
nabla/nn/layers/normalization.py
nabla/nn/layers/pooling.py
nabla/nn/losses/__init__.py
nabla/nn/losses/classification.py
nabla/nn/losses/regression.py
nabla/nn/losses/regularization.py
nabla/nn/optim/__init__.py
nabla/nn/optim/adam.py
nabla/nn/optim/adamw.py
nabla/nn/optim/base.py
nabla/nn/optim/schedules.py
nabla/nn/optim/sgd.py
nabla/nn/utils/__init__.py
nabla/nn/utils/checkpointing.py
nabla/nn/utils/metrics.py
nabla/nn/utils/parameters.py
nabla/nn/utils/training.py
nabla/ops/__init__.py
nabla/ops/binary.py
nabla/ops/conv_utils.py
nabla/ops/creation.py
nabla/ops/linalg.py
nabla/ops/operation.py
nabla/ops/reduce.py
nabla/ops/unary.py
nabla/ops/view.py
nabla/ops/kernels/__init__.mojo
nabla/utils/__init__.py
nabla/utils/broadcasting.py
nabla/utils/formatting.py
nabla/utils/max_interop.py
nabla/utils/types.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
scripts/generate_api_docs.py
scripts/release.py
scripts/update_docs.py
scripts/yank_old_releases.py