jaxfit (pypi): Uses JAX for GPU/TPU accelerated nonlinear least-squares curve fitting
Keywords: fit|curve-fitting|gpu-acceleration|jax|least-squares|nonlinear-regression|optimization|python