jaxfit
Curve Fitter
A package for GPU/TPU accelerated nonlinear least-squares curve fitting using JAX
GPU/TPU accelerated nonlinear least-squares curve fitting using JAX
53 stars
1 watching
5 forks
Language: Python
last commit: about 3 years agoLinked from 1 awesome list
curve-fittinggpu-accelerationjaxleast-squaresnonlinear-regressionoptimizationpython
n2cholas/awesome-jaxRelated projects:
| Repository | Description | Stars |
|---|---|---|
 vindaar/nim-mpfit | A wrapper for the cMPFIT library to perform non-linear least squares fitting in programming languages. | 17 |
 google/jaxopt | An open-source project providing hardware accelerated, batchable and differentiable optimizers in JAX for deep learning. | 941 |
 ott-jax/ott | A JAX-based library for computing optimal transport problems in various settings | 550 |
 google-deepmind/kfac-jax | Library providing an implementation of the K-FAC optimizer and curvature estimator for second-order optimization in neural networks. | 252 |
 nvidia/jax-toolbox | Provides optimized tools and infrastructure for JAX development on NVIDIA GPUs | 268 |
 dfm/extending-jax | This project provides infrastructure to interface custom C++ and CUDA code with the JAX library for scientific computing | 379 |
 jaxgaussianprocesses/gpjax | Provides a low-level interface to Gaussian process models in JAX for flexible extension and customisation | 467 |
 snopt/snopt-matlab | Provides a Matlab interface to SNOPT for nonlinear optimization | 56 |
 patrick-kidger/diffrax | Provides numerical differential equation solvers using autodifferentiable and GPU-capable JAX. | 1,480 |
 locuslab/optnet | A PyTorch module that adds differentiable optimization as a layer to neural networks | 517 |
 dynamiqs/dynamiqs | A high-performance quantum systems simulation library using JAX on GPUs | 184 |
 huntermcgushion/hyperparameter_hunter | Automates hyperparameter optimization and result saving across machine learning algorithms | 706 |
| locuslab/qpth | A solver for optimizing quadratic programs in deep learning and other fields. | 691 |
 ethz-adrl/ifopt | An Eigen-based C++ interface to nonlinear programming solvers | 790 |
 jeremiecoullon/jax-tqdm | Library that enhances JAX with support for dynamic progress bars in scans and loops | 100 |