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
51 stars
1 watching
4 forks
Language: Python
last commit: over 1 year ago
Linked 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. | 933 |
 ott-jax/ott | A JAX-based library for computing optimal transport problems in various settings | 525 |
 google-deepmind/kfac-jax | Library providing an implementation of the K-FAC optimizer and curvature estimator for second-order optimization in neural networks. | 248 |
 nvidia/jax-toolbox | A collection of optimized JAX libraries and examples for simplified development on NVIDIA GPUs | 245 |
 dfm/extending-jax | This project provides infrastructure to interface custom C++ and CUDA code with the JAX library for scientific computing | 378 |
 jaxgaussianprocesses/gpjax | Provides a low-level interface to Gaussian process models in JAX for flexible extension and customisation | 461 |
 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,442 |
 locuslab/optnet | A PyTorch module that adds differentiable optimization as a layer to neural networks | 513 |
 dynamiqs/dynamiqs | A high-performance quantum systems simulation library built with JAX. | 161 |
 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. | 685 |
 ethz-adrl/ifopt | An Eigen-based C++ interface to nonlinear programming solvers | 787 |
 jeremiecoullon/jax-tqdm | Library that enhances JAX with support for dynamic progress bars in scans and loops | 96 |