GPU/TPU accelerated nonlinear least-squares curve fitting using JAX
☆62Jun 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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Generative models with JAX.☆21Feb 21, 2026Updated last month
- Object-oriented Jax framework extending Equinox for scientific programming☆25Mar 20, 2026Updated 3 weeks ago
- AMIDE: a Medical Imaging Data Examiner☆26Feb 6, 2026Updated 2 months ago
- Jax implementation of the AdaHessian optimizer☆20Mar 11, 2021Updated 5 years ago
- ☆13Jul 12, 2024Updated last year
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- Schedule free optimiser implemented in JAX using Optimistix☆15May 29, 2024Updated last year
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆565Apr 5, 2026Updated last week
- Interactive textbook on state space models☆12Apr 11, 2022Updated 4 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
- A miscellaneous set of helper functions, custom distributions, and other utilities that I find useful when using NumPyro in my work☆28Feb 1, 2025Updated last year
- Deep memory and sequence models in JAX☆25Jan 15, 2026Updated 2 months ago
- A zoo of implementations of differential equation problems in NumPy and JAX. Oscillators, chemical reactions, n-body problems, epidemiolo…☆17Dec 28, 2023Updated 2 years ago
- Turn SymPy expressions into trainable JAX expressions.☆366Feb 4, 2026Updated 2 months ago
- NordVPN Special Discount Offer • AdSave on top-rated NordVPN 1 or 2-year plans with secure browsing, privacy protection, and support for for all major platforms.
- Code for the paper: Kernel Distributionally Robust Optimization☆13Feb 21, 2021Updated 5 years ago
- Use numba-compiled kernels from within Jax☆32May 1, 2025Updated 11 months ago
- FALCON is a Python-based software application designed to facilitate PET motion correction, both for head and total-body scans. Our pro…☆48Mar 26, 2026Updated 2 weeks ago
- Proximal Nested Sampling for high-dimensional Bayesian model selection☆26Apr 18, 2024Updated last year
- An HMC/NUTS implementation in Aesara☆31Jun 22, 2023Updated 2 years ago
- Minimal Implementation of Bayesian Optimization in JAX☆105Apr 24, 2025Updated 11 months ago
- Normalizing-flow enhanced sampling package for probabilistic inference in Jax☆256Oct 28, 2025Updated 5 months ago
- Automatic mesh generation.☆32Updated this week
- Julia interface to hypre linear solvers☆32Apr 1, 2026Updated last week
- DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Julia package for fast flexible adaptive rejection sampling for log-concave univariate densities.☆19Jul 25, 2023Updated 2 years ago
- A Wadler–Lindig pretty printer for Python☆46Apr 5, 2026Updated last week
- PyTorch implementation of Bezier simplex fitting☆12Updated this week
- Transpile from Julia to Python☆17Mar 26, 2026Updated 2 weeks ago
- An introduction to functional programming for scalable statistical computing☆78Jul 2, 2024Updated last year
- ☆280Mar 24, 2026Updated 2 weeks ago
- Model predictive path integral control in jax☆17Feb 8, 2026Updated 2 months ago
- It's not ROS... but it is ROS-y! Simple, fast inter-process message passing for distributed Python processes☆11Mar 30, 2026Updated last week
- Differentiable Spline Approximations☆14Sep 3, 2023Updated 2 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Exponential families for JAX☆77Updated this week
- Tools for JAX☆50Apr 2, 2026Updated last week
- Comparsion of Julia's GPU Kernel based ODE solvers with other open-source GPU ODE solvers☆28Jan 4, 2024Updated 2 years ago
- Mathematical operations for JAX pytrees☆210Dec 5, 2024Updated last year
- BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.☆1,044Updated this week
- 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