Dipolar-Quantum-Gases / jaxfitLinks
GPU/TPU accelerated nonlinear least-squares curve fitting using JAX
☆58Updated 2 years ago
Alternatives and similar repositories for jaxfit
Users that are interested in jaxfit are comparing it to the libraries listed below
Sorting:
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆132Updated last year
- A small library of paramaterizations and parameter constraints for PyTrees.☆45Updated 5 months ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆103Updated last week
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆53Updated last month
- Matrix-free linear algebra in JAX.☆145Updated last month
- Interpolation and function approximation with JAX☆226Updated this week
- Probabilistic programming framework for signal inference algorithms that operate regardless of the underlying grids and their resolutions☆37Updated last week
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆74Updated last month
- Numerical quadrature with JAX☆72Updated this week
- Python Algorithms for Randomized Linear Algebra☆56Updated 2 years ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆58Updated 3 months ago
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆65Updated 3 weeks ago
- S2FFT: Differentiable and accelerated spherical transforms☆208Updated last week
- Minimal Implementation of Bayesian Optimization in JAX☆98Updated 6 months ago
- Stencil computations in JAX☆71Updated 2 years ago
- Differentiable and gpu enabled fast wavelet transforms in JAX.☆46Updated last year
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆209Updated 2 months ago
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆116Updated 3 weeks ago
- Optimal numerical differentiation of noisy time series data in python.☆66Updated 3 months ago
- Multiple dispatch over abstract array types in JAX.☆131Updated 2 weeks ago
- Methods for numerical differentiation of noisy data in python☆126Updated last week
- Add a tqdm progress bar to your JAX scans and loops.☆118Updated 5 months ago
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆50Updated 3 months ago
- A library to define abstract linear operators, and associated algebra and matrix-free algorithms, that works with pyTorch Tensors.☆15Updated 8 months ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆481Updated last week
- Orthogonal polynomials with JAX☆26Updated last week
- Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"☆30Updated last year
- Mathematical operations for JAX pytrees☆201Updated 10 months ago
- Github Action to: Check version / Test / git tag / GitHub Release / Deploy to PyPI☆20Updated 3 weeks ago