Add WeatherNext-style ONNX conversion demo - #469
Add WeatherNext-style ONNX conversion demo#469justinchuby with Copilot wants to merge 16 commits into
Conversation
|
|
Signed-off-by: GitHub <noreply@github.com> Co-authored-by: justinchuby <11205048+justinchuby@users.noreply.github.com>
Signed-off-by: GitHub <noreply@github.com> Co-authored-by: justinchuby <11205048+justinchuby@users.noreply.github.com>
Signed-off-by: GitHub <noreply@github.com> Co-authored-by: justinchuby <11205048+justinchuby@users.noreply.github.com>
Signed-off-by: GitHub <noreply@github.com> Co-authored-by: justinchuby <11205048+justinchuby@users.noreply.github.com>
Signed-off-by: GitHub <noreply@github.com> Co-authored-by: justinchuby <11205048+justinchuby@users.noreply.github.com>
Signed-off-by: GitHub <noreply@github.com> Co-authored-by: justinchuby <11205048+justinchuby@users.noreply.github.com>
Signed-off-by: GitHub <noreply@github.com> Co-authored-by: justinchuby <11205048+justinchuby@users.noreply.github.com>
|
@copilot what I wanted was a formal support of this model in the modeling/integation files, and a demo script that can run this model on real data. |
Co-authored-by: justinchuby <11205048+justinchuby@users.noreply.github.com>
Co-authored-by: justinchuby <11205048+justinchuby@users.noreply.github.com>
Co-authored-by: justinchuby <11205048+justinchuby@users.noreply.github.com>
Co-authored-by: justinchuby <11205048+justinchuby@users.noreply.github.com>
Co-authored-by: justinchuby <11205048+justinchuby@users.noreply.github.com>
Co-authored-by: justinchuby <11205048+justinchuby@users.noreply.github.com>
Co-authored-by: justinchuby <11205048+justinchuby@users.noreply.github.com>
Co-authored-by: justinchuby <11205048+justinchuby@users.noreply.github.com>
Implemented in |
WeatherNext uses a JAX/Haiku + xarray architecture that does not map to the existing HuggingFace model registry flow. This PR adds a runnable ONNX demo that sketches the WeatherNext one-step forecast contract and graph workflow under
examples/.Demo workflow
examples/weathernext.py.ModelPackage.--validate.WeatherNext-style graph contract
input_state: weather variables on a lat/lon gridforcings: known future forcing variablessample_noise: stochastic FGN-style noise channelsnext_state: one-step forecast gridCoverage
tests/weathernext_example_test.pyto verify the exported ONNX graph I/O contract.Example: