AIbase
Product LibraryTool Navigation

ivon-optax

Public

An Optax-based JAX implementation of the IVON optimizer for large-scale VI training of NNs (ICML'24 spotlight)

Creat2024-07-29T22:37:25
Update2025-02-10T17:08:42
5
Stars
0
Stars Increase