neuralgcm/neuralgcm issues and pull requests

#247 - Noisy Geopotential from stochastic model

Issue - State: open - Opened by ShihengDuan 2 days ago

#246 - Added `self.mesh: parallelism.Mesh` attribute to api.ForecastSystem. Added `from_fiddle_config` class-methods that support updating parallelism settings when instantiating a model. Added fiddle_tags.py to support modular updates to Mesh instances when using multiple Mesh components in a more complex models.

Pull Request - State: closed - Opened by copybara-service[bot] 3 days ago

#245 - Added CoordinateShard to coordinates.py; Included spmd_mesh attribute to LonLatGrid and SphericalHarmonicGrid; Added an option to passing additional coord_types to Field.from_xarray."

Pull Request - State: closed - Opened by copybara-service[bot] 3 days ago

#244 - [coordax] Make coordinates for named dimensions optional

Pull Request - State: closed - Opened by copybara-service[bot] 6 days ago

#243 - Fix pytype inconsistency in api.py

Pull Request - State: closed - Opened by copybara-service[bot] 6 days ago

#242 - [neuralgcm] Explicitly set radius=1.0 for neuralgcm v1

Pull Request - State: closed - Opened by copybara-service[bot] 6 days ago

#241 - Is it necessary for me to install the google_proprietary_code library to run the experiment.py script in reference_code folder?

Issue - State: open - Opened by Steven04453X 7 days ago

#240 - Add coordax.tag and coordax.untag

Pull Request - State: closed - Opened by copybara-service[bot] 9 days ago

#239 - Use coordax to replace neuralgcm.experimental.xarray_utils

Pull Request - State: closed - Opened by copybara-service[bot] 9 days ago

#238 - Use JAX-Datetime instead of sim_time in NeuralGCM experimental

Pull Request - State: closed - Opened by copybara-service[bot] 10 days ago

#237 - Split coordax/core.py into fields.py and coordinate_systems.py

Pull Request - State: closed - Opened by copybara-service[bot] 10 days ago

#236 - Improve validation in coordax.Field

Pull Request - State: closed - Opened by copybara-service[bot] 10 days ago

#235 - Add jax_datetime and jax_solar into neuralgcm/experimental

Pull Request - State: closed - Opened by copybara-service[bot] 10 days ago

#234 - Fix coordax.xarray_test.test_data_array_to_field_no_match on GitHub CI

Pull Request - State: closed - Opened by copybara-service[bot] 13 days ago

#233 - Added chex.set_n_cpu to conftest to fix issues with running tests that simulate multiple devices.

Pull Request - State: closed - Opened by copybara-service[bot] 13 days ago

#232 - Tweak names for NeuralGCM CI builds

Pull Request - State: closed - Opened by copybara-service[bot] 13 days ago

#231 - Remove coordax.is_positional_prefix_field().

Pull Request - State: closed - Opened by copybara-service[bot] 14 days ago

#230 - Update Coordax Xarray interface

Pull Request - State: closed - Opened by copybara-service[bot] 14 days ago

#229 - Coordax coordinate matching from Xarray for NeuralGCM coordinates

Pull Request - State: open - Opened by copybara-service[bot] 14 days ago

#229 - Coordax coordinate matching from Xarray for NeuralGCM coordinates

Pull Request - State: closed - Opened by copybara-service[bot] 14 days ago

#228 - Added parallelism.Mesh objects to model components that implement sharding constraints for model parallelism similar to the old API.

Pull Request - State: closed - Opened by copybara-service[bot] 14 days ago

#227 - Added with_era5_levels constructors to Pressure levels.

Pull Request - State: closed - Opened by copybara-service[bot] 14 days ago

#227 - Added with_era5_levels constructors to Pressure levels.

Pull Request - State: closed - Opened by copybara-service[bot] 14 days ago

#226 - Added temporary helper functions with_physics_sharding with_dycore_sharding to support model parallelism before we switch to using coordax.Field for intermediate representations.

Pull Request - State: closed - Opened by copybara-service[bot] 15 days ago

#225 - WIP: Coordax coordinate matching from Xarray for NeuralGCM

Pull Request - State: closed - Opened by copybara-service[bot] 16 days ago

#224 - NeuralGCM release: dinosaur -> dinosaur-dycore

Pull Request - State: closed - Opened by copybara-service[bot] 16 days ago

#223 - Relax required precision in stochastic tests

Pull Request - State: closed - Opened by copybara-service[bot] 16 days ago

#222 - Correct InstantDiagnostic initialization.

Pull Request - State: closed - Opened by copybara-service[bot] 16 days ago

#222 - Correct InstantDiagnostic initialization.

Pull Request - State: closed - Opened by copybara-service[bot] 16 days ago

#221 - Drop dependency of coordax.Coordinate on penzai.struct

Pull Request - State: closed - Opened by copybara-service[bot] 20 days ago

#220 - Added parallelism.py that implements a helper object that stores sharding strategies and an spmd_mesh. and simplifies adding sharding constraints to arrays and coordax.Field objects.

Pull Request - State: closed - Opened by copybara-service[bot] 20 days ago

#219 - Prototype xarray converters for coordax

Pull Request - State: closed - Opened by copybara-service[bot] 21 days ago

#218 - Delete stray properties from a bad merge.

Pull Request - State: closed - Opened by copybara-service[bot] 22 days ago

#217 - Add .axes property to coordax.Coordinates

Pull Request - State: closed - Opened by copybara-service[bot] 22 days ago

#216 - Fix typo TakeSurfaceAdjacentSgimaLevel -> TakeSurfaceAdjacentSigmaLevel

Pull Request - State: closed - Opened by copybara-service[bot] 22 days ago

#215 - Adding precipitation minus evaporation transform that could be calculated as a diagnostic.

Pull Request - State: closed - Opened by copybara-service[bot] 22 days ago

#214 - [coordax] replace penzai.NamedArray with coordax NamedArray

Pull Request - State: closed - Opened by copybara-service[bot] 23 days ago

#213 - [coordax] Support ellipsis (...) in NamedArray.tag

Pull Request - State: closed - Opened by copybara-service[bot] 24 days ago

#212 - [coordax.named_axes] make pytree flattening more robust

Pull Request - State: closed - Opened by copybara-service[bot] 24 days ago

#211 - Add coordax.named_axes.NamedArray grad test

Pull Request - State: closed - Opened by copybara-service[bot] 27 days ago

#210 - Nudging capability in the current version

Issue - State: open - Opened by ShihengDuan 27 days ago - 2 comments

#209 - Fixed bug in experimental xarray_utils that due to non-determinism of set can result in error reconstructing coordinates.

Pull Request - State: closed - Opened by copybara-service[bot] 27 days ago

#208 - Added embeddings and tests.

Pull Request - State: open - Opened by copybara-service[bot] 28 days ago

#207 - Eliminate usages of `CompositeCheckpointHandler` that use `item_names` or `items_and_handlers`.

Pull Request - State: closed - Opened by copybara-service[bot] 29 days ago

#206 - Use JAX-Datetime instead of sim_time in NeuralGCM experimental

Pull Request - State: closed - Opened by copybara-service[bot] 29 days ago

#205 - Add named_axes.tag() and untag() functions.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 month ago

#204 - No public description

Pull Request - State: open - Opened by copybara-service[bot] about 1 month ago - 1 comment

#203 - Small edits to deepdive_into_models.ipynb

Pull Request - State: closed - Opened by copybara-service[bot] about 1 month ago

#202 - Bump jinja2 from 3.1.4 to 3.1.5 in /docs

Pull Request - State: closed - Opened by dependabot[bot] about 1 month ago - 1 comment
Labels: dependencies

#201 - [dinosaur] remove StateWithTime

Pull Request - State: closed - Opened by copybara-service[bot] about 1 month ago

#200 - [neural-gcms] handle missing sim_time values.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 month ago

#199 - Added components and tests for U-Net tower.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 month ago - 1 comment

#198 - ValueError: Configurable 'BatchGaussianRandomFieldModule' doesn't have a parameter named 'n_fixed_fields'

Issue - State: closed - Opened by Steven04453X about 1 month ago - 2 comments

#197 - Hangups after unrolling

Issue - State: open - Opened by anthony-meza about 1 month ago - 1 comment

#196 - How to update with your new paper:1 output the precipitation instead of p_e;2.set the ensemble run ?

Issue - State: open - Opened by weatherforecasterwhai about 2 months ago - 2 comments

#195 - No public description

Pull Request - State: closed - Opened by copybara-service[bot] about 2 months ago

#194 - Copy-editing of NeuralGCM dataset docs

Pull Request - State: closed - Opened by copybara-service[bot] about 2 months ago

#193 - Add detailed information about NeuralGCM datasets that are available.

Pull Request - State: closed - Opened by copybara-service[bot] about 2 months ago

#192 - Would you please explain the Transformer and attention mentioned in neuralgcm codes?

Issue - State: closed - Opened by weatherforecasterwhai about 2 months ago - 1 comment

#192 - Would you please explain the Transformer and attention mentioned in neuralgcm codes?

Issue - State: closed - Opened by weatherforecasterwhai about 2 months ago - 1 comment

#191 - Add detailed information about pre-trained models to the NeuralGCM docs.

Pull Request - State: closed - Opened by copybara-service[bot] about 2 months ago

#191 - Add detailed information about pre-trained models to the NeuralGCM docs.

Pull Request - State: closed - Opened by copybara-service[bot] about 2 months ago

#190 - DLL load failed while importing utils

Issue - State: closed - Opened by Detachment73 about 2 months ago - 1 comment

#189 - Release NeuralGCM 1.1.0

Pull Request - State: closed - Opened by copybara-service[bot] about 2 months ago

#188 - 04_30_2024 .pkl file is uncompatible with latest neuralgcm and dinosaur version

Issue - State: closed - Opened by weatherforecasterwhai about 2 months ago - 8 comments

#187 - Migrate from jax.core to jax.extend.core for several deprecated symbols

Pull Request - State: closed - Opened by copybara-service[bot] about 2 months ago

#186 - Can't download neural_gcm_dynamic_forcing_deterministic_2_8_deg.pkl by gsutil anymore

Issue - State: closed - Opened by weatherforecasterwhai about 2 months ago - 2 comments