esa / torchquad
Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX
☆189Updated 4 months ago
Related projects ⓘ
Alternatives and complementary repositories for torchquad
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆94Updated 2 months ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆332Updated this week
- Interpolation and function approximation with JAX☆135Updated 3 weeks ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆121Updated 2 months ago
- GPU/TPU accelerated nonlinear least-squares curve fitting using JAX☆52Updated last year
- Turning SymPy expressions into PyTorch modules.☆142Updated last year
- Matrix-free linear algebra in JAX.☆106Updated 2 months ago
- Differentiable and accelerated spherical transforms with JAX☆136Updated this week
- Code for the paper "Learning Differential Equations that are Easy to Solve"☆270Updated 2 years ago
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆360Updated last month
- A parallel ODE solver for PyTorch☆232Updated last month
- Gaussian processes in JAX.☆461Updated 2 weeks ago
- Newton and Quasi-Newton optimization with PyTorch☆320Updated 8 months ago
- Turn SymPy expressions into trainable JAX expressions.☆322Updated 7 months ago
- ☆107Updated 3 weeks ago
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆249Updated this week
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆445Updated 2 weeks ago
- Use scipy.optimize.minimize as a PyTorch Optimizer.☆65Updated 3 months ago
- Mathematical operations for JAX pytrees☆189Updated 6 months ago
- Probabilistic Numerics in Python.☆439Updated 6 months ago
- Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"☆170Updated 2 years ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆80Updated last week
- Python Algorithms for Randomized Linear Algebra☆46Updated last year
- Multiple dispatch over abstract array types in JAX.☆105Updated last week
- ☆409Updated last month
- Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy☆75Updated 2 weeks ago
- Add a tqdm progress bar to your JAX scans and loops.☆96Updated 3 weeks ago
- A differentiable cosmology library in JAX☆178Updated 3 months ago
- Numerical quadrature with JAX☆44Updated 3 weeks ago
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆89Updated last year