Release v1.0

[suvayu]

General requirements

Compatibility with Jax and the numerical ecosystem (diffrax, equinox, ...)

Maintenance tasks

• Change package setup to match modern convetions #13

Required features

• A robust config system #8
• Flexible data loader #4
• Improved training framework #2
• Training modes: full time-series or single time step predition #3

Optional features

• Provide a small number of pre-trained models #5
• Compatibility with other native codes #6
• HPC integration via SLURM for hyperparameter optimisation #7