rsokl / MyGradLinks
Drop-in autodiff for NumPy.
☆211Updated 6 months 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☆198Updated 7 months ago
- Documentation:☆120Updated 2 years ago
- Stencil computations in JAX☆71Updated last year
- Tools for JAX☆47Updated 3 weeks ago
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- A suite of benchmarks for CPU and GPU performance of the most popular high-performance libraries for Python☆328Updated 9 months ago
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆484Updated last week
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆128Updated 9 months ago
- Probabilistic Numerics in Python.☆452Updated last week
- A dead simple Python package for creating custom JAX pytree objects☆63Updated 8 months ago
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆205Updated 7 months ago
- Exponential families for JAX☆72Updated 3 weeks ago
- Turn SymPy expressions into trainable JAX expressions.☆343Updated 2 months ago
- Extending JAX with custom C++ and CUDA code☆398Updated 10 months ago
- Multiple dispatch over abstract array types in JAX.☆125Updated 3 weeks ago
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆269Updated this week
- Create Numpy .npy files by appending on the growth axis☆63Updated 2 weeks ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆111Updated 4 months ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆97Updated last week
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆45Updated 6 months ago
- Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX☆100Updated 2 years ago
- Solve automatic numerical differentiation problems in one or more variables.☆266Updated 2 years ago
- Minimal Implementation of Bayesian Optimization in JAX☆95Updated 2 months ago
- Turning SymPy expressions into PyTorch modules.☆148Updated last year
- The tiniest of Gaussian Process libraries☆313Updated this week
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆437Updated last month
- Probabilistic Programming and Nested sampling in JAX☆187Updated last week
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆97Updated last year
- GPU/TPU accelerated nonlinear least-squares curve fitting using JAX☆57Updated 2 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago