Skip to main content

Samudra

Open Quickstart in Colab 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.

Sea Surface Temperature of the Tropical Pacific: Ground Truth vs Samudra v2

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

To configure, train, validate, and visualize Samudra 2 in a browser on public data with a Colab GPU, open the free-tier Colab quickstart.

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.1.0.tar.gz (30.1 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.1.0-py3-none-any.whl (249.9 kB view details)

Uploaded Python 3

File details

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

File metadata

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

File hashes

Hashes for samudra-0.1.0.tar.gz
Algorithm Hash digest
SHA256 b563ba01d4bc1fe82099a026cc3ebffab8587d8817c1c9b075946b65f786d16f
MD5 62a4ac72f0278fc47729ac20e9c0d578
BLAKE2b-256 2af429947ac182fc53f45731e5c95dca14fb24f7977174f82188157016d12080

See more details on using hashes here.

Provenance

The following attestation bundles were made for samudra-0.1.0.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.1.0-py3-none-any.whl.

File metadata

  • Download URL: samudra-0.1.0-py3-none-any.whl
  • Upload date:
  • Size: 249.9 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.1.0-py3-none-any.whl
Algorithm Hash digest
SHA256 1e5771ce86c324d816d927260c2d0d2f41c0884c8dd6bacfaa350f41b33ba4df
MD5 5ab636e11faee7ae4daa9ace45138b0d
BLAKE2b-256 15ddcedd4fa8395ffe2b293edea92cd14a462aa5ff13c66dddddb442b18ca7a1

See more details on using hashes here.

Provenance

The following attestation bundles were made for samudra-0.1.0-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