GPU/TPU accelerated nonlinear least-squares curve fitting using JAX
☆61Jun 23, 2023Updated 2 years ago
Alternatives and similar repositories for jaxfit
Users that are interested in jaxfit are comparing it to the libraries listed below
Sorting:
- Generative models with JAX.☆21Feb 21, 2026Updated 3 weeks ago
- Object-oriented Jax framework extending Equinox for scientific programming☆25Mar 10, 2026Updated last week
- ☆12Oct 8, 2020Updated 5 years ago
- Jax implementation of the AdaHessian optimizer☆20Mar 11, 2021Updated 5 years ago
- ☆13Jul 12, 2024Updated last year
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆554Mar 9, 2026Updated last week
- Schedule free optimiser implemented in JAX using Optimistix☆15May 29, 2024Updated last year
- Interactive textbook on state space models☆12Apr 11, 2022Updated 3 years ago
- Stencil computations in JAX☆71Nov 15, 2025Updated 4 months ago
- A dead simple Python package for creating custom JAX pytree objects☆66Nov 14, 2024Updated last year
- Jax Decompiler☆16Apr 22, 2025Updated 10 months ago
- Deep memory and sequence models in JAX☆23Jan 15, 2026Updated 2 months ago
- Code for "Disentangling images with Lie group transformations and sparse coding" (2023).☆13May 24, 2021Updated 4 years ago
- A miscellaneous set of helper functions, custom distributions, and other utilities that I find useful when using NumPyro in my work☆27Feb 1, 2025Updated last year
- Python code for manipulating data from the SXS collaboration☆33Mar 11, 2026Updated last week
- Turn SymPy expressions into trainable JAX expressions.☆364Feb 4, 2026Updated last month
- Code for the paper: Kernel Distributionally Robust Optimization☆13Feb 21, 2021Updated 5 years ago
- Use numba-compiled kernels from within Jax☆32May 1, 2025Updated 10 months ago
- ☆13Feb 28, 2017Updated 9 years ago
- An HMC/NUTS implementation in Aesara☆31Jun 22, 2023Updated 2 years ago
- Minimal Implementation of Bayesian Optimization in JAX☆104Apr 24, 2025Updated 10 months ago
- Normalizing-flow enhanced sampling package for probabilistic inference in Jax☆256Oct 28, 2025Updated 4 months ago
- Automatic mesh generation.☆32Updated this week
- FALCON is a Python-based software application designed to facilitate PET motion correction, both for head and total-body scans. Our pro…☆49Dec 2, 2025Updated 3 months ago
- Julia interface to hypre linear solvers☆32Jan 21, 2026Updated 2 months ago
- Julia package for fast flexible adaptive rejection sampling for log-concave univariate densities.☆19Jul 25, 2023Updated 2 years ago
- PyTorch implementation of Bezier simplex fitting☆12Updated this week
- Transpile from Julia to Python☆17Mar 7, 2026Updated last week
- ☆275Updated this week
- Napari plugin for custom analysis and visualization of lattice lightsheet and Oblique Plane Microscopy data. The plugin is optimized for …☆14Updated this week
- It's not ROS... but it is ROS-y! Simple, fast inter-process message passing for distributed Python processes☆11Mar 12, 2026Updated last week
- Exponential families for JAX☆76Mar 13, 2026Updated last week
- Tools for JAX☆50Mar 13, 2026Updated last week
- BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.☆1,033Feb 3, 2026Updated last month
- Differentiable non-uniform interpolation: https://arxiv.org/abs/2012.13257☆10Oct 3, 2021Updated 4 years ago
- Mathematical operations for JAX pytrees☆209Dec 5, 2024Updated last year
- Code for the "Unified Modulated Pattern Analysis" (UMPA) method of processing speckle-based X-ray imaging data☆12Aug 28, 2024Updated last year
- A haptic authoring tool☆11May 19, 2025Updated 10 months ago
- A python/Pytorch re-implementation of several classical Magnetic Resonance Imaging (MRI) reconstruction algorithms☆39Nov 28, 2023Updated 2 years ago