GPU/TPU accelerated nonlinear least-squares curve fitting using JAX
☆61Jun 23, 2023Updated 3 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.☆22Feb 21, 2026Updated 4 months ago
- ☆12Oct 8, 2020Updated 5 years ago
- AMIDE: a Medical Imaging Data Examiner☆26Feb 6, 2026Updated 4 months ago
- Jax implementation of the AdaHessian optimizer☆19Mar 11, 2021Updated 5 years ago
- ☆13Jul 12, 2024Updated last year
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Schedule free optimiser implemented in JAX using Optimistix☆15May 29, 2024Updated 2 years ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆598May 13, 2026Updated last month
- Interactive textbook on state space models☆12Apr 11, 2022Updated 4 years ago
- Stencil computations in JAX☆72Nov 15, 2025Updated 7 months ago
- A dead simple Python package for creating custom JAX pytree objects☆66Nov 14, 2024Updated last year
- Jax Decompiler☆16Apr 22, 2025Updated last year
- Code for "Disentangling images with Lie group transformations and sparse coding" (2023).☆13May 24, 2021Updated 5 years ago
- 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
- A zoo of implementations of differential equation problems in NumPy and JAX. Oscillators, chemical reactions, n-body problems, epidemiolo…☆18Dec 28, 2023Updated 2 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Python code for manipulating data from the SXS collaboration☆33Jun 17, 2026Updated 2 weeks ago
- Turn SymPy expressions into trainable JAX expressions.☆369Apr 20, 2026Updated 2 months ago
- Code for the paper: Kernel Distributionally Robust Optimization☆13Feb 21, 2021Updated 5 years ago
- This repository is the demo implementation of [Deep Dimension Reduction for Supervised Representation Learning].☆12Sep 12, 2024Updated last year
- Deep memory and sequence models in JAX☆32Jun 8, 2026Updated 3 weeks ago
- Use numba-compiled kernels from within Jax☆32May 1, 2025Updated last year
- Numeric Typst☆27May 10, 2026Updated last month
- Proximal Nested Sampling for high-dimensional Bayesian model selection☆27Apr 18, 2024Updated 2 years ago
- Minimal Implementation of Bayesian Optimization in JAX☆108Apr 24, 2025Updated last year
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- An HMC/NUTS implementation in Aesara☆31Jun 22, 2023Updated 3 years ago
- Normalizing-flow enhanced sampling package for probabilistic inference in Jax☆259Oct 28, 2025Updated 8 months ago
- Automatic mesh generation.☆35Updated this week
- A Wadler–Lindig pretty printer for Python☆48Apr 20, 2026Updated 2 months ago
- PyTorch implementation of Bezier simplex fitting☆12Jun 18, 2026Updated 2 weeks ago
- An introduction to functional programming for scalable statistical computing☆80Jul 2, 2024Updated 2 years ago
- Transpile from Julia to Python☆17Jun 18, 2026Updated 2 weeks ago
- ☆303Updated this week
- Napari plugin for custom analysis and visualization of lattice lightsheet and Oblique Plane Microscopy data. The plugin is optimized for …☆15Jun 24, 2026Updated last week
- AI Agents on 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.
- It's not ROS... but it is ROS-y! Simple, fast inter-process message passing for distributed Python processes☆11May 29, 2026Updated last month
- Differentiable Spline Approximations☆14Sep 3, 2023Updated 2 years ago
- Exponential families for JAX☆77Jun 26, 2026Updated last week
- Tools for JAX☆50Jun 26, 2026Updated last week
- Comparsion of Julia's GPU Kernel based ODE solvers with other open-source GPU ODE solvers☆27Jan 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,091Jun 23, 2026Updated last week