Skip to main content
Pre-release

This release is a pre-release and may not be stable for production use.

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

Metadata

Release files for samudra 0.1.2.dev202610191144

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for samudra 0.1.2.dev202610191144
File Size Uploaded Provenance Actions
samudra-0.1.2.dev202610191144.tar.gz 30.3 MB With provenance Details for samudra-0.1.2.dev202610191144.tar.gz

Built distribution (wheel)

Table of built distributions (wheels) for samudra 0.1.2.dev202610191144
File Interpreter ABI Platform Actions
samudra-0.1.2.dev202610191144-py3-none-any.whl Python 3 none any Details for samudra-0.1.2.dev202610191144-py3-none-any.whl

Total release size: 30.6 MB

Release files / samudra-0.1.2.dev202610191144.tar.gz

Download URL samudra-0.1.2.dev202610191144.tar.gz
Size 30.3 MB
Tags Source
SHA-256 checksum
How to use checksums
47439d9082e25b9681e5592aa0d2a90471c8cb2b554797ff9697955314b320ca
BLAKE2b-256 checksum
How to use checksums
c9316ad3c63d7543678ae83fe1daa9cc44ff758626666dcbebb8ad936fd704a2
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via twine/6.1.0 CPython/3.13.14

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

PyPI Publish Attestation

PyPI verified that this artifact, at this checksum, originated from the publisher listed below.

Signed by GitHub Actions, verified by PyPI on Oct 19, 2026.

Transparency log

Release files / samudra-0.1.2.dev202610191144-py3-none-any.whl

Download URL samudra-0.1.2.dev202610191144-py3-none-any.whl
Size 310.0 kB
Tags Python 3
SHA-256 checksum
How to use checksums
56e187ba283325eb793de9697ef7bd61bb06ebed098c437965e2f3926be4a22c
BLAKE2b-256 checksum
How to use checksums
7431f069e6edf75ab781779d247119449f1dc6fea091b0df362172fd0de92483
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via twine/6.1.0 CPython/3.13.14

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

PyPI Publish Attestation

PyPI verified that this artifact, at this checksum, originated from the publisher listed below.

Signed by GitHub Actions, verified by PyPI on Oct 19, 2026.

Transparency log

Security for samudra 0.1.2.dev202610191144

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

Full provenance

Both files in this release have been uploaded with provenance metadata.

Select from 2 files

Attestation for samudra-0.1.2.dev202610191144.tar.gz

PyPI Publish Attestation

PyPI verified that this artifact, at this checksum, originated from the publisher listed below.

Signed by GitHub Actions, verified by PyPI on Oct 19, 2026.

Transparency log

A note on package security

PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.

To learn more about protecting yourself from malicious actors, visit our package security documentation.

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