rsokl / MyGrad
Drop-in autodiff for NumPy.
☆204Updated 2 months ago
Alternatives and similar repositories for MyGrad:
Users that are interested in MyGrad are comparing it to the libraries listed below
- A simple tool for logging and plotting measurements during machine learning experiments☆38Updated 2 years ago
- Mathematical operations for JAX pytrees☆198Updated 3 months ago
- Turn SymPy expressions into trainable JAX expressions.☆331Updated 2 months ago
- Documentation:☆119Updated last year
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆386Updated last month
- Probabilistic Numerics in Python.☆448Updated 10 months ago
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆469Updated last week
- Turning SymPy expressions into JAX functions☆44Updated 4 years ago
- Tools for JAX☆45Updated last week
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆107Updated 2 weeks ago
- Extending JAX with custom C++ and CUDA code☆388Updated 7 months ago
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆201Updated 4 months ago
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆251Updated this week
- Likelihood-free AMortized Posterior Estimation with PyTorch☆126Updated 7 months ago
- Stencil computations in JAX☆70Updated last year
- Exponential families for JAX☆63Updated this week
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆426Updated last week
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆123Updated 6 months ago
- Create Numpy .npy files by appending on the growth axis☆63Updated 6 months ago
- The tiniest of Gaussian Process libraries☆307Updated last week
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- Minimal Implementation of Bayesian Optimization in JAX☆90Updated 2 months ago
- A dead simple Python package for creating custom JAX pytree objects☆63Updated 4 months ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆102Updated last year
- Turning SymPy expressions into PyTorch modules.☆146Updated last year
- Probabilistic Programming and Nested sampling in JAX☆172Updated 3 months ago
- numba_scipy extends Numba to make it aware of SciPy☆269Updated 6 months ago
- Add a tqdm progress bar to your JAX scans and loops.☆104Updated 5 months ago
- Simplicial Homology Global Optimization☆49Updated 2 weeks ago
- Gaussian processes in JAX.☆487Updated this week