rsokl / MyGrad
Drop-in autodiff for NumPy.
☆196Updated 2 months ago
Related projects ⓘ
Alternatives and complementary repositories for MyGrad
- A simple tool for logging and plotting measurements during machine learning experiments☆37Updated last year
- Mathematical operations for JAX pytrees☆189Updated 6 months ago
- Extending JAX with custom C++ and CUDA code☆378Updated 3 months ago
- Documentation:☆120Updated last year
- Multiple dispatch over abstract array types in JAX.☆106Updated last week
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆333Updated last week
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆446Updated 3 weeks ago
- Stencil computations in JAX☆66Updated last year
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆95Updated 2 months ago
- Minimal Implementation of Bayesian Optimization in JAX☆84Updated 6 months ago
- Probabilistic Numerics in Python.☆439Updated 6 months ago
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆219Updated last week
- The tiniest of Gaussian Process libraries☆297Updated 2 weeks ago
- Pure Python/NumPy neural network library extending MyGrad☆21Updated 7 months ago
- Tools for JAX☆43Updated last week
- A small library for creating and manipulating custom JAX Pytree classes☆57Updated last year
- Exponential families for JAX☆56Updated this week
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆121Updated 2 months ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆81Updated 2 weeks ago
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆365Updated last month
- Turning SymPy expressions into PyTorch modules.☆142Updated last year
- ☆110Updated 3 weeks ago
- A dead simple Python package for creating custom JAX pytree objects☆62Updated last week
- Probabilistic Programming and Nested sampling in JAX☆150Updated last week
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆96Updated last year
- Abstract your array operations.☆137Updated last month
- Normalizing Flows using JAX☆82Updated 11 months ago
- Add a tqdm progress bar to your JAX scans and loops.☆96Updated last month
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆250Updated this week
- Interpolation and function approximation with JAX☆137Updated 3 weeks ago