mattjj / autodidactView external linksLinks
A pedagogical implementation of Autograd
☆1,012May 26, 2020Updated 5 years ago
Alternatives and similar repositories for autodidact
Users that are interested in autodidact are comparing it to the libraries listed below
Sorting:
- Efficiently computes derivatives of NumPy code.☆7,455Updated this week
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆1,026Dec 17, 2025Updated last month
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,771Updated this week
- Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more☆34,848Updated this week
- Research language for array processing in the Haskell/ML family☆1,659Jan 5, 2026Updated last month
- Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.☆2,600Updated this week
- JAX-based neural network library☆3,185Jan 30, 2026Updated 2 weeks ago
- Source-to-Source Debuggable Derivatives in Pure Python☆2,316Sep 29, 2022Updated 3 years ago
- Optax is a gradient processing and optimization library for JAX.☆2,177Updated this week
- JAX - A curated list of resources https://github.com/google/jax☆2,041Jan 20, 2026Updated 3 weeks ago
- ☆922Jan 29, 2026Updated 2 weeks ago
- Flax is a neural network library for JAX that is designed for flexibility.☆7,066Feb 7, 2026Updated last week
- higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…☆1,629Mar 25, 2022Updated 3 years ago
- Fast and Easy Infinite Neural Networks in Python☆2,374Mar 1, 2024Updated last year
- Extending JAX with custom C++ and CUDA code☆403Aug 18, 2024Updated last year
- Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/☆1,898Updated this week
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,395Jan 26, 2026Updated 2 weeks ago
- Normalizing Flows in Jax☆107Aug 19, 2020Updated 5 years ago
- functorch is JAX-like composable function transforms for PyTorch.☆1,436Aug 21, 2025Updated 5 months ago
- Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.☆6,338Apr 4, 2025Updated 10 months ago
- ☆622Jan 23, 2026Updated 3 weeks ago
- Exponential families for JAX☆76Feb 1, 2026Updated last week
- A highly efficient implementation of Gaussian Processes in PyTorch☆3,828Feb 5, 2026Updated last week
- A generic Monte Carlo method based on the Gumbel-Max trick.☆32May 31, 2016Updated 9 years ago
- ☆775Jan 27, 2024Updated 2 years ago
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆311Feb 6, 2026Updated last week
- Differentiable SDE solvers with GPU support and efficient sensitivity analysis.☆1,705Dec 30, 2024Updated last year
- Understanding computation in artificial and biological recurrent networks through the lens of dynamical systems.☆390May 28, 2021Updated 4 years ago
- AlgoPy is a Research Prototype for Algorithmic Differentation in Python☆88Jul 7, 2024Updated last year
- Code for the paper "Learning Differential Equations that are Easy to Solve"☆286Nov 25, 2021Updated 4 years ago
- ☆153May 25, 2020Updated 5 years ago
- Deep universal probabilistic programming with Python and PyTorch☆8,971Jul 9, 2025Updated 7 months ago
- ☆50Oct 22, 2020Updated 5 years ago
- KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows☆1,156Feb 6, 2026Updated last week
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆298Jan 9, 2026Updated last month
- Probabilistic Torch is library for deep generative models that extends PyTorch☆894May 12, 2024Updated last year
- A Graph Neural Network Library in Jax☆1,463Mar 18, 2024Updated last year
- A probabilistic programming language in TensorFlow. Deep generative models, variational inference.☆4,840Mar 18, 2024Updated last year
- A tiny scalar-valued autograd engine and a neural net library on top of it with PyTorch-like API☆14,598Aug 8, 2024Updated last year