GitHub repository: simudt/GradientAscent-Jax
Primary language: Python
Created: 2024-03-10T20:25:37Z
Description: Custom gradient ascent solver (optimizer) for JAX/Flax models