Dipolar-Quantum-Gases / jaxfitView external linksLinks
GPU/TPU accelerated nonlinear least-squares curve fitting using JAX
☆59Jun 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.☆20Jan 2, 2025Updated last year
- Stencil computations in JAX☆70Nov 15, 2025Updated 3 months ago
- ☆13Jul 12, 2024Updated last year
- 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
- Deep memory and sequence models in JAX☆23Jan 15, 2026Updated last month
- Jax Decompiler☆16Apr 22, 2025Updated 9 months ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆541Feb 9, 2026Updated last week
- A dead simple Python package for creating custom JAX pytree objects☆66Nov 14, 2024Updated last year
- Schedule free optimiser implemented in JAX using Optimistix☆15May 29, 2024Updated last year
- Object-oriented Jax framework extending Equinox for scientific programming☆24Dec 16, 2025Updated 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
- Exponential families for JAX☆76Feb 9, 2026Updated last week
- Jax implementation of the AdaHessian optimizer☆20Mar 11, 2021Updated 4 years ago
- Mathematical operations for JAX pytrees☆209Dec 5, 2024Updated last year
- An introduction to functional programming for scalable statistical computing☆78Jul 2, 2024Updated last year
- A Wadler–Lindig pretty printer for Python☆44Jan 16, 2026Updated last month
- Julia package for fast flexible adaptive rejection sampling for log-concave univariate densities.☆19Jul 25, 2023Updated 2 years ago
- Helpful functions for use alongside the rich Python library.☆20Aug 26, 2025Updated 5 months ago
- Turn SymPy expressions into trainable JAX expressions.☆360Feb 4, 2026Updated last week
- Proximal Nested Sampling for high-dimensional Bayesian model selection☆26Apr 18, 2024Updated last year
- Tools for JAX☆50Jan 21, 2026Updated 3 weeks ago
- ☆265Feb 9, 2026Updated last week
- Minimal Implementation of Bayesian Optimization in JAX☆102Apr 24, 2025Updated 9 months ago
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆21Sep 18, 2020Updated 5 years ago
- Comparsion of Julia's GPU Kernel based ODE solvers with other open-source GPU ODE solvers☆28Jan 4, 2024Updated 2 years ago
- Julia implementation of ADMM solver on multiple GPUs☆23Updated this week
- Probabilistic Programming and Nested sampling in JAX☆221Dec 15, 2025Updated 2 months ago
- A CVXPY extension for saddle problems☆28Jan 1, 2026Updated last month
- ☆55Aug 9, 2024Updated last year
- Code for "Continuous-Time Meta-Learning with Forward Mode Differentiation" (ICLR 2022)☆25Mar 4, 2022Updated 3 years ago
- This repository contains PyTorch implementations of various random feature maps for dot product kernels.☆22Jul 13, 2024Updated last year
- The tiniest of Gaussian Process libraries☆333Feb 8, 2026Updated last week
- Use numba-compiled kernels from within Jax☆31May 1, 2025Updated 9 months ago
- Manage your Julia dependencies from Python☆65Jan 19, 2026Updated 3 weeks ago
- ☆67Aug 21, 2024Updated last year
- Forward mode laplacian implemented in JAX tracer☆30Jan 7, 2026Updated last month
- The official repository of Decoupled Reinforcement Learning to Stabilise Intrinsically-Motivated Exploration" (AAMAS 2022)☆27Feb 3, 2022Updated 4 years ago
- Normalizing-flow enhanced sampling package for probabilistic inference in Jax☆255Oct 28, 2025Updated 3 months ago
- Add a tqdm progress bar to your JAX scans and loops.☆124May 9, 2025Updated 9 months ago