jaxfit
GPU/TPU accelerated nonlinear least-squares curve fitting using JAX
AI summary
Curve Fitter
A package for GPU/TPU accelerated nonlinear least-squares curve fitting using JAX
- stars
- 53
- forks
- 5
- watching
- 1
- awesome list
- 1
Similar projects
Found by comparing what the projects do, not just their names.
Fitter
A wrapper for the cMPFIT library to perform non-linear least squares fitting in programming languages.
Deep learning optimizer
An open-source project providing hardware accelerated, batchable and differentiable optimizers in JAX for deep learning.
ott-jax/ott550
Optimal matcher
A JAX-based library for computing optimal transport problems in various settings
KFAC optimizer
Library providing an implementation of the K-FAC optimizer and curvature estimator for second-order optimization in neural networks.
JAX toolbox
Provides optimized tools and infrastructure for JAX development on NVIDIA GPUs
JAX wrapper
This project provides infrastructure to interface custom C++ and CUDA code with the JAX library for scientific computing
GP model library
Provides a low-level interface to Gaussian process models in JAX for flexible extension and customisation
Optimizer library
Provides a Matlab interface to SNOPT for nonlinear optimization
Differential equation solver
Provides numerical differential equation solvers using autodifferentiable and GPU-capable JAX.
Optimization layer
A PyTorch module that adds differentiable optimization as a layer to neural networks
Quantum simulator
A high-performance quantum systems simulation library using JAX on GPUs
Optimizer
Automates hyperparameter optimization and result saving across machine learning algorithms
QP optimizer
A solver for optimizing quadratic programs in deep learning and other fields.
Nonlinear solver interface
An Eigen-based C++ interface to nonlinear programming solvers
JAX progress bar library
Library that enhances JAX with support for dynamic progress bars in scans and loops
Featured in 1 awesome list
Each link jumps to the spot where the list mentions jaxfit.
