Machine Learning Model TrainingML Frameworks

Jaxopt

Hardware accelerated, batchable and differentiable optimizers in JAX.

Open Source

About

JAXopt is a library designed for optimization tasks, offering hardware-accelerated and differentiable optimizers that can run on GPU, TPU, and CPU. It allows for batch processing of multiple instances of the same optimization problem through JAX's vectorization capabilities. The library supports implicit differentiation and automatic differentiation of optimization solutions, making it suitable for users in deep learning and differentiable programming. However, it is important to note that JAXopt is no longer maintained or developed.

Open Source Health

Not enough history
Stars
1,062
Forks
76
License
Apache-2.0
Last commit
1 months ago
Python

Related Categories