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.0.3.dev202608030929

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.0.3.dev202608030929
File Size Uploaded Provenance Actions
samudra-0.0.3.dev202608030929.tar.gz 30.0 MB With provenance Details for samudra-0.0.3.dev202608030929.tar.gz

Built distribution (wheel)

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

Total release size: 30.2 MB

Release files / samudra-0.0.3.dev202608030929.tar.gz

Download URL samudra-0.0.3.dev202608030929.tar.gz
Size 30.0 MB
Tags Source
SHA-256 checksum
How to use checksums
890feec3a497ceb928cfec104adc3284f4ce6b8841fd308fdfbd6c28d65f6742
BLAKE2b-256 checksum
How to use checksums
c99555571df7ca4c7072e0c43f8bf8e46e51fc60398281b7625f9bb337426d06
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 Aug 3, 2026.

Transparency log

Release files / samudra-0.0.3.dev202608030929-py3-none-any.whl

Download URL samudra-0.0.3.dev202608030929-py3-none-any.whl
Size 193.6 kB
Tags Python 3
SHA-256 checksum
How to use checksums
530b0a8bc7ed6919ae0e8c909937b4087444a6743715faa5958340fe252a9de4
BLAKE2b-256 checksum
How to use checksums
f3e6b86fbda25de1e594eea3808645814cf313170863af586f734baba8db38b7
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 Aug 3, 2026.

Transparency log

Security for samudra 0.0.3.dev202608030929

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.0.3.dev202608030929.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 Aug 3, 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