All pull requests

Pull requests

• Add opt-in pad_pool to Samudra so no row or column is dropped by pooling

  #1578·adam-subel opened Oct 12, 2026Contributor·1 of 2 tasks·

  ·
  1

• Add output_dir to inference configs to decouple job dir from data output dir

  #1576·yyexela opened Oct 12, 2026Contributor·1 of 2 tasks·

  ·

• Add Muon optimizer option to OptimizationConfig

  #1574·elynnwu opened Oct 9, 2026Contributor·1 of 2 tasks·

  ·
  1

• Per-level pool padding for Samudra (pad_pool), lat_pad naming, and pole continuation past the far pole

  #1573·adam-subel opened Oct 9, 2026Contributor·1 of 2 tasks·

  ·

• (don't merge) Add 1° 5-daily sea-ice budget companion stores to the OM4 pipeline

  #1565·jpdunc23 opened Oct 7, 2026Member·1 of 2 tasks·

  ·
  2

• Add mld_wright97 and pbo_wright97 ocean derived variables

  #1563·jpdunc23 opened Oct 6, 2026Member·1 of 2 tasks·

  ·
  3

• Update UFS Pipeline

  #1560·Arcomano1234 opened Oct 5, 2026Contributor·2 tasks·

  ·

• Add a prescribed_cell_mean surface energy flux correction with a checkpointed runoff heat map

  #1559·elynnwu opened Oct 3, 2026Contributor·1 of 2 tasks·

  ·
  4

• Replace fsspec with obstore for remote I/O in fme

  #1553·oliverwm1 opened Oct 2, 2026Collaborator·1 of 2 tasks·

  ·
  5

• Add pushforward_steps option to TrainStepperConfig

  #1551·mcgibbon opened Oct 2, 2026Contributor·1 of 2 tasks·

  ·
  2

• Make Samudra and m2lines CappedGELU traceable by torch.compile

  #1549·yyexela opened Oct 1, 2026Contributor·1 of 2 tasks·

  ·
  2

• Deduplicate stepper override application into a shared helper

  #1547·mcgibbon opened Oct 1, 2026Contributor·1 of 2 tasks·

  ·
  5

• Pass the network normalizer to GMR and input masking at call time

  #1545·yyexela opened Sep 30, 2026Contributor·1 of 2 tasks·

  ·
  3

• Add patch_size, num_levels and skip_projection options to the Swin U-Net

  #1544·yyexela opened Sep 30, 2026Contributor·2/2·1 of 2 tasks·

  ·
  4

• Swin U-Net attention and conditioning performance

  #1543·yyexela opened Sep 30, 2026Contributor·1/2·1 of 2 tasks·

  ·
  3

• Build unconditional modules as if the dataset carried no labels

  #1541·yyexela opened Sep 30, 2026Contributor·1/2·1 of 2 tasks·

  ·
  3

• Add float32_matmul_precision option to SingleModuleStepConfig

  #1539·yyexela opened Sep 30, 2026Contributor·1 of 2 tasks·

  ·
  3

• Add get_pooled_stats.py with per-group stats output

  #1537·yyexela opened Sep 30, 2026Contributor·1 of 2 tasks·

  ·
  7

• Add per-group network normalization

  #1536·yyexela opened Sep 30, 2026Contributor·2/2·1 of 2 tasks·

  ·
  10

• Add inference compile override

  #1527·yyexela opened Sep 28, 2026Contributor·1 of 2 tasks·

  ·
  3

• Recover wandb metrics lost to preemption from on-disk logs

  #1526·yyexela opened Sep 28, 2026Contributor·1 of 2 tasks·

  ·
  32

• Fix SecondaryDecoder.wrap_module mutating the decoder in place

  #1520·yyexela opened Sep 23, 2026Contributor·1 of 2 tasks·

  ·
  1

• Make irfft, NullTimer.child, CappedGELU, and envelope clipping traceable by torch.compile

  #1519·yyexela opened Sep 23, 2026Contributor·1 of 2 tasks·

  ·
  10

• Add per-component corrector_loss to the coupled train stepper

  #1515·jpdunc23 opened Sep 22, 2026Member·1 of 2 tasks·

  ·

• Extend torch.compile to every step type and module builder

  #1514·yyexela opened Sep 21, 2026Contributor·1 of 2 tasks·

  ·
  9