CITATION.cff
MANIFEST.in
README.md
pyproject.toml
readme.md
docs/conf.py
docs/conf_new.py
docs/conf_old.py
docs/verify_seo.py
examples/add_one_custom.py
examples/conds_jit.py
examples/debug_epoch_8.py
examples/debug_sjit_nan.py
examples/debug_vjp_issue.py
examples/jit_analysis.py
examples/jvp_examples.py
examples/mlp_builder_showcase.py
examples/mlp_inference.py
examples/mlp_inference_jit.py
examples/mlp_train.py
examples/mlp_train_jax.py
examples/mlp_train_jax_jit.py
examples/mlp_train_jit.py
examples/mlp_train_modular.py
examples/mlp_train_sjit.py
examples/mlp_train_sjit_working.py
examples/test.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/conv_utils.py
nabla/ops/creation.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/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/generate_api_docs.py
scripts/release.py
scripts/update_docs.py
scripts/yank_old_releases.py