weathernext
No description available.
File Explorer
- demo_gridding_and_tracking.ipynb
- cloud_vm_setup.md
- GenCast_0p25deg_accelerator_scorecard.png
- GenCast_0p25deg_attention_implementation_scorecard.png
- GenCast_1p0deg_Mini_ENS_scorecard.png
- gencast_demo_cloud_vm.ipynb
- gencast_mini_demo.ipynb
- local_runtime_popup_1.png
- local_runtime_popup_2.png
- local_runtime_url.png
- project.png
- provision_tpu.png
- README.md
- tpu_types.png
- graphcast_demo.ipynb
- README.md
- wn2_demo.ipynb
- __init__.py
- constants.py
- cyclone_utils.py
- data_pipeline_utils.py
- direct_tracker.py
- direct_tracker_6h_v1_config.py
- ibtracs_netcdf_to_csv.py
- ibtracs_processing_stages.py
- ibtracs_processing_utils.py
- tracker_base.py
- tracker_utils.py
- __init__.py
- deep_typed_graph_net.py
- grid_mesh_connectivity.py
- grid_mesh_connectivity_test.py
- __init__.py
- activations.py
- autoregressive.py
- casting.py
- checkpoint.py
- checkpoint_test.py
- data_modalities.py
- data_utils.py
- data_utils_test.py
- deep_gnn.py
- dense.py
- ensemble.py
- fiddle_config_io.py
- gather_scatter_ops.py
- icosahedral_mesh.py
- icosahedral_mesh_test.py
- losses.py
- mesh_transformer.py
- model_utils.py
- nan_cleaning.py
- normalization.py
- padding_utils.py
- points_mesh_gnn.py
- predictor_base.py
- rollout.py
- sharding.py
- sharding_utils.py
- solar_radiation.py
- solar_radiation_test.py
- sparse_transformer.py
- sparse_transformer_utils.py
- task.py
- typed_graph.py
- typed_graph_net.py
- update_blocks.py
- update_blocks_utils.py
- variables.py
- xarray_dense.py
- xarray_tree.py
- xarray_tree_test.py
- __init__.py
- denoiser.py
- denoisers_base.py
- dpm_solver_plus_plus_2s.py
- gencast.py
- samplers_base.py
- samplers_utils.py
- transformer.py
- __init__.py
- graphcast.py
- WeatherNext2.json
- WeatherNextCyclones.json
- WeatherNextCyclones_Mini.json
- __init__.py
- architecture.py
- architecture_utils.py
- fgn.py
- __init__.py
- CONTRIBUTING.md
- LICENSE
- README.md
- setup.py
# Installation Guide
pythonPrerequisites
- Python >=3.10
- JAX / TPU / GPU Runtime Latest
Setup Steps
pip install git+https://github.com/google-deepmind/[email protected]
Install the WeatherNext package directly from the official GitHub repository.
Key Commands
pip install git+https://github.com/google-deepmind/[email protected]
Install WeatherNext library and its dependencies.
Running on TPU or H100/P100 GPU environments is recommended, and model weights must be downloaded from the Google Cloud Storage bucket.
# Use via CDN
jsDelivrjsDelivr serves any public GitHub repository as a CDN with zero setup. Pick a version and a file to get a ready-to-paste link and snippet.
Link
Example
Command Glossary
Commands referenced in this DOCs, explained below.
pip install
View Details ▼
pip install
Install Python packages.
pip install {{package1 package2 ...}}
Install one or more packages:
pip install {{package1 package2 ...}} {{[-U|--upgrade]}}
Upgrade all specified packages to the latest version, installing any that are not already present:
pip install {{package}}=={{version}}
Install a specific version of a package:
// repository documentation
Was this content helpful?
(0 ratings)