JAX implementation of Competitive Gradient Descent algorithms
Project description
JAX-CGD
JAX implementation of Competitive Gradient Descent algorithms for minimax optimization problems.
Features
- ACGD (Adaptive Competitive Gradient Descent): An adaptive variant of competitive gradient descent with momentum
- BCGD (Basic Competitive Gradient Descent): Standard competitive gradient descent algorithm
- Flexible Solvers: Support for different linear algebra solvers (CG, GMRES)
- JAX Integration: Built on JAX for automatic differentiation and JIT compilation
Installation
pip install jax-cgd
Algorithms
ACGD (Adaptive Competitive Gradient Descent)
- Includes momentum and adaptive learning rates
- Better convergence properties for complex minimax problems
- Configurable exponential decay rate (beta) and numerical stability (eps)
BCGD (Basic Competitive Gradient Descent)
- Standard competitive gradient descent
- Simpler implementation for basic minimax optimization
Requirements
- Python >= 3.10
- JAX >= 0.5.1
- NumPy
- Flax >= 1.10.0
License
MIT License
Contributing
Contributions are welcome! Please feel free to submit a Pull Request.
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
jax_cgd-1.0.1.tar.gz
(8.2 kB
view details)
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file jax_cgd-1.0.1.tar.gz.
File metadata
- Download URL: jax_cgd-1.0.1.tar.gz
- Upload date:
- Size: 8.2 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.0.1 CPython/3.10.16
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
9a2672cffb1a4e1d43db22592aabdb0f16138ea596e4ff97cec04da780efd2d3
|
|
| MD5 |
271c2a567df74b4de72686127eb4d884
|
|
| BLAKE2b-256 |
d065d97b92969a9b31d061e8532b8ab97f3efb3660e15b0df0e733235f48988d
|
File details
Details for the file jax_cgd-1.0.1-py3-none-any.whl.
File metadata
- Download URL: jax_cgd-1.0.1-py3-none-any.whl
- Upload date:
- Size: 9.0 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.0.1 CPython/3.10.16
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
6de8e129f66e4ec1e5fe79ab1efbe9fdb74879cf5259e7bab707c47dcf365d20
|
|
| MD5 |
af11d4ff9ee8caa7518f58dd246d11cd
|
|
| BLAKE2b-256 |
64ac39e7bd143b53826a22b53539e7c7f9354974ed16522e57b4f36cf70a9534
|
