rasmusbergpalm / nanograd
A miniscule implementation of reverse mode auto-differentiation
☆27Updated 3 years ago
Related projects: ⓘ
- 🦠 AD in less than 20 lines☆48Updated 3 years ago
- ☆56Updated 2 years ago
- Documentation:☆119Updated last year
- Minimal Implementation of Bayesian Optimization in JAX☆82Updated 4 months ago
- Mathematical operations for JAX pytrees☆181Updated 4 months ago
- Dive into Jax, Flax, XLA and C++☆31Updated 4 years ago
- Code for the book "The Elements of Differentiable Programming".☆57Updated 3 weeks ago
- A small library for creating and manipulating custom JAX Pytree classes☆59Updated last year
- Inference Combinators in JAX☆42Updated 2 months ago
- Run PyTorch in JAX. 🤝☆187Updated 10 months ago
- Tools for JAX☆44Updated this week
- A functional training loops library for JAX☆84Updated 7 months ago
- ☆97Updated 2 months ago
- Turn jitted jax functions back into python source code☆20Updated 2 months ago
- Multiple dispatch over abstract array types in JAX.☆100Updated last month
- Bayesian algorithm execution (BAX)☆46Updated 3 years ago
- ☆78Updated 3 years ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆119Updated this week
- Pytorch implementation of SuperPolyak subgradient method.☆44Updated last year
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆39Updated 11 months ago
- Exponential families for JAX☆54Updated last week
- Graph Learning with JAX☆14Updated 2 years ago
- Turning SymPy expressions into JAX functions☆42Updated 3 years ago
- A Python package of computer vision models for the Equinox ecosystem.☆100Updated 2 months ago
- Named Tensors for Legible Deep Learning in JAX☆147Updated this week
- Stencil computations in JAX☆66Updated 11 months ago
- An introduction to functional programming for scalable statistical computing☆69Updated 2 months ago
- OpTree: Optimized PyTree Utilities☆136Updated this week
- Exact OU processes with JAX☆30Updated last month
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆34Updated 3 years ago