Samudra

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.

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.1.dev202609071112.tar.gz (30.2 MB view details)

Uploaded Sep 7, 2026 Source

Built Distribution

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

The dropdown lists show the available interpreters, ABIs, and platforms. Enable javascript to be able to filter the list of wheel files.

samudra-0.1.1.dev202609071112-py3-none-any.whl (286.1 kB view details)

Uploaded Sep 7, 2026 Python 3

File details

Details for the file samudra-0.1.1.dev202609071112.tar.gz.

File metadata

Download URL: samudra-0.1.1.dev202609071112.tar.gz
Upload date: Sep 7, 2026
Size: 30.2 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.1.dev202609071112.tar.gz
Algorithm	Hash digest
SHA256	`3b746c79c2a2e742cc51acdc32af5b54177c20b9ab8c54894723421b56d8ef02`
MD5	`f3a66edf2ab3b4e78a7d51169592e55a`
BLAKE2b-256	`87249402d143d0fa12cc730fee4ebb69f315911df0b8801c2309eaa5c42e7e3e`

See more details on using hashes here.

Provenance

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

Statement:
- Statement type: https://in-toto.io/Statement/v1
- Predicate type: https://docs.pypi.org/attestations/publish/v1
- Subject name: samudra-0.1.1.dev202609071112.tar.gz
- Subject digest: 3b746c79c2a2e742cc51acdc32af5b54177c20b9ab8c54894723421b56d8ef02
- Sigstore transparency entry: 2750021142
- Sigstore integration time: Sep 7, 2026
Source repository:
- Permalink: m2lines/Samudra@55fe01dde9739b1967ecfe9b81a074ef0a61e19a
- Branch / Tag: refs/heads/main
- Owner: https://github.com/m2lines
- Access: public
Publication detail:
- Token Issuer: https://token.actions.githubusercontent.com
- Runner Environment: github-hosted
- Publication workflow: release.yml@55fe01dde9739b1967ecfe9b81a074ef0a61e19a
- Trigger Event: schedule