This CL introduces a JAX/Flax training script (`trainer.py`) for the Global MetNet model, including functions for model initialization, optimization, loss computation, and the training loop. The README is updated with system requirements, installation instructions, and guidance on using the provided code to train the model.
6c149caShreya Agrawal4mo ago