Stax

MaRDI QIDQ53771swMATHFDO

Official website https://jax.readthedocs.io/en/latest/jax.example_libraries.stax.html

Source code repository https://github.com/jax-ml/jax/blob/main/jax/example_libraries/stax.py

Cited in

(3)

• Multilevel Fine-Tuning: Closing Generalization Gaps in Approximation of Solution Maps under a Limited Budget for Training Data
• neural-tangents
• Haiku

This page was built for software: Stax