GitHub repository: Ceyron/trainax
Primary language: Python
Description: Training methodologies for autoregressive neural operators in JAX.