rsokl / MyGradLinks
Drop-in autodiff for NumPy.
☆213Updated 3 weeks ago
Alternatives and similar repositories for MyGrad
Users that are interested in MyGrad are comparing it to the libraries listed below
Sorting:
- Mathematical operations for JAX pytrees☆202Updated 11 months ago
- Stencil computations in JAX☆71Updated 2 years ago
- Turn SymPy expressions into trainable JAX expressions.☆353Updated 6 months ago
- Documentation:☆125Updated 2 years ago
- Probabilistic Numerics in Python.☆459Updated 4 months ago
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆209Updated 3 months ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆132Updated last year
- Tools for JAX☆48Updated 2 weeks ago
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- Multiple dispatch over abstract array types in JAX.☆136Updated this week
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆284Updated last month
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆498Updated last month
- Extending JAX with custom C++ and CUDA code☆401Updated last year
- Exponential families for JAX☆75Updated 2 weeks ago
- Probabilistic Programming and Nested sampling in JAX☆211Updated this week
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆45Updated 10 months ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆116Updated last week
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆489Updated this week
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆492Updated last week
- Turning SymPy expressions into PyTorch modules.☆152Updated 2 years ago
- A dead simple Python package for creating custom JAX pytree objects☆65Updated last year
- GPU/TPU accelerated nonlinear least-squares curve fitting using JAX☆58Updated 2 years ago
- Minimal Implementation of Bayesian Optimization in JAX☆100Updated 6 months ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆106Updated last week
- A suite of benchmarks for CPU and GPU performance of the most popular high-performance libraries for Python☆333Updated last year
- The tiniest of Gaussian Process libraries☆324Updated last week
- Solve automatic numerical differentiation problems in one or more variables.☆276Updated last month
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆104Updated 2 years ago
- Python Algorithms for Randomized Linear Algebra☆56Updated 2 years ago
- Abstract your array operations.☆160Updated 2 weeks ago