Skip to main content

Samudra

Pre-commit Test CPU Benchmark CPU Benchmark Results

Samudra is a global ocean emulator described in "Samudra: An AI Global Ocean Emulator for Climate" and updated in "Samudra 2: Scaling Ocean Emulators across Resolutions". Samudra efficiently emulates the ocean component of a state-of-the-art climate model, accurately reproducing key ocean variables including sea surface height, horizontal velocities, temperature, and salinity, across their full depth.

Ground truth (left) vs. Samudra 2 prediction (right) for sea surface temperature in the tropical Pacific.

We are actively and openly developing this emulator to support new tasks and data sources with the goal of building a broadly useful foundation model for ocean and climate. Please see our docs for more or our contributing guide to join in!

How to run

Given uv is installed:

# Launch a training run with the default Samudra configuration
uvx samudra train samudra_om4/train.yaml --data @data/om4_demo.yaml

# Run a long autoregressive rollout against ground-truth data, collecting metrics.
uvx samudra eval samudra_om4/eval.yaml --data @data/om4_demo.yaml

# Generate maps, time series, and probability density plots from eval outputs.
uvx samudra viz samudra_om4/viz.yaml --data @data/om4_demo.yaml

To learn more, check out the installation and quick start guides.

NYU      Princeton University      Open Athena

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

samudra-0.0.2.tar.gz (30.0 MB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

samudra-0.0.2-py3-none-any.whl (197.1 kB view details)

Uploaded Python 3

File details

Details for the file samudra-0.0.2.tar.gz.

File metadata

  • Download URL: samudra-0.0.2.tar.gz
  • Upload date:
  • Size: 30.0 MB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.14

File hashes

Hashes for samudra-0.0.2.tar.gz
Algorithm Hash digest
SHA256 1c04ac5457678d82449030ff78944bc92001f88d0ba904c9888da418dbfa5bef
MD5 c95e65e7c432c43ccdaf473d13c3219f
BLAKE2b-256 9ee69a241249a5e38cbf3dd98e4097d84cea560981fca0f105b10a25e00ad4a2

See more details on using hashes here.

Provenance

The following attestation bundles were made for samudra-0.0.2.tar.gz:

Publisher: release.yml on m2lines/Samudra

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file samudra-0.0.2-py3-none-any.whl.

File metadata

  • Download URL: samudra-0.0.2-py3-none-any.whl
  • Upload date:
  • Size: 197.1 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.14

File hashes

Hashes for samudra-0.0.2-py3-none-any.whl
Algorithm Hash digest
SHA256 1a796ed12662a541a54359aa25cedc5dd8a5ffcd2e91756b9fc4605857303a67
MD5 333f8780fcc3f981ae2fbbbd5df054c9
BLAKE2b-256 bc4f7cccdc024967e7c1f3c8e7f35f7d7727e8e6161b521bdfec0158ec0e5954

See more details on using hashes here.

Provenance

The following attestation bundles were made for samudra-0.0.2-py3-none-any.whl:

Publisher: release.yml on m2lines/Samudra

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.
Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page