CITATION.cff
MANIFEST.in
README.md
pyproject.toml
readme.md
docs/conf.py
docs/scripts/check_indexing.py
docs/scripts/configure_seo.py
docs/scripts/final_seo_report.py
docs/scripts/fix_duplicate_viewport.py
docs/scripts/fix_seo_issues.py
docs/scripts/generate_api_docs.py
docs/scripts/generate_sitemap.py
docs/scripts/seo_audit.py
docs/scripts/seo_improvements.py
examples/add_one_custom.py
examples/conds_djit.py
examples/jit_performance_test.py
examples/jvp_examples.py
examples/mlp_train.py
examples/mlp_train_djit.py
examples/mlp_train_gpu.py
examples/mlp_train_jax.py
examples/mlp_train_jax_jit.py
examples/mlp_train_jit.py
examples/nodoc_decorator_usage.py
examples/nonlinear_regression.py
examples/spiral_classification.py
examples/transformer_train.py
examples/transformer_train_jax.py
examples/transformer_train_jax_enhanced.py
examples/transformer_train_jax_jit.py
examples/transformer_train_jit.py
examples/transformer_train_pytorch.py
examples/value_and_grads_gpu.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/nn/__init__.py
nabla/nn/architectures/__init__.py
nabla/nn/architectures/mlp.py
nabla/nn/init/__init__.py
nabla/nn/init/variance_scaling.py
nabla/nn/layers/__init__.py
nabla/nn/layers/activations.py
nabla/nn/layers/linear.py
nabla/nn/losses/__init__.py
nabla/nn/losses/classification.py
nabla/nn/losses/regression.py
nabla/nn/optim/__init__.py
nabla/nn/optim/adam.py
nabla/nn/optim/adamw.py
nabla/nn/optim/schedules.py
nabla/nn/optim/sgd.py
nabla/nn/utils/__init__.py
nabla/nn/utils/metrics.py
nabla/nn/utils/regularization.py
nabla/nn/utils/training.py
nabla/ops/__init__.py
nabla/ops/binary.py
nabla/ops/creation.py
nabla/ops/indexing.py
nabla/ops/linalg.py
nabla/ops/operation.py
nabla/ops/reduce.py
nabla/ops/special.py
nabla/ops/unary.py
nabla/ops/view.py
nabla/ops/kernels/__init__.mojo
nabla/transforms/__init__.py
nabla/transforms/grad.py
nabla/transforms/jacfwd.py
nabla/transforms/jacrev.py
nabla/transforms/jit.py
nabla/transforms/jvp.py
nabla/transforms/utils.py
nabla/transforms/vjp.py
nabla/transforms/vmap.py
nabla/utils/__init__.py
nabla/utils/docs.py
nabla/utils/formatting.py
nabla/utils/grad_utils.py
nabla/utils/max_interop.py
nabla/utils/shape_utils.py
nabla/utils/testing.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/release.py