Skip to main content

⬡ Kontinua

The Open-Core Physics AI Platform

Train, evaluate, and deploy neural PDE surrogates 1,000× faster than traditional HPC solvers.


License: Apache-2.0 PyPI Docs arXiv NeurIPS HuggingFace


⚡ What is Kontinua?

Kontinua is an open-core physics AI platform built for engineers, researchers, and scientific ML practitioners. It replaces expensive, multi-day HPC simulation runs (CFD, FEA, MHD) with millisecond-scale neural operator and transformer surrogates.

🌟 Core Capabilities

  • Open-Source Core (pip install kontinua): 8 production-ready architectures (FNO, CNextU-Net, TFNO, AViT, AFNO, DilatedResNet, ReFNO), standardized spatial/spectral/temporal evaluation metrics, physics-aware normalization, and local ONNX/TensorRT export.
  • 15TB Multi-Physics Corpus: Native streaming and distributed loaders for 16 physical domains from The Well (NeurIPS 2024), covering fluid dynamics, astrophysics, acoustic scattering, viscoelasticity, and biological active matter.
  • Kontinua Cloud: Auto-scaling NVIDIA Triton GPU inference endpoints, sub-50ms latency REST/gRPC APIs, managed domain fine-tuning pipelines, and private benchmark CI/CD.

🚀 Quickstart

Installation

pip install kontinua

For benchmark evaluation and hardware acceleration:

pip install kontinua[benchmark,gpu]

1. Run Local Neural Surrogate Inference

import torch
from kontinua.core.models import CNextUNet
from kontinua.core.data import WellDataset

# Load pre-trained surrogate checkpoint
model = CNextUNet.from_pretrained("kontinua/cnext-unet-turbulence")
model.eval()

# Load initial condition / boundary states
dataset = WellDataset(well_base_path="./data", well_dataset_name="turbulence", well_split_name="test")
sample = dataset[0]["input"].unsqueeze(0)

# Generate 100-step spatiotemporal prediction in milliseconds
with torch.no_grad():
    prediction = model(sample)

print(f"Prediction shape: {prediction.shape}")

2. Connect to Kontinua Cloud API

import kontinua

# Authenticate with your Kontinua Cloud API key
client = kontinua.Client(api_key="ko_live_...")

# Execute cloud inference on managed Triton GPU cluster
result = client.predict(
    domain="turbulence",
    input_path="./boundary_condition.hdf5",
    model="cnext-unet-v2",
    rollout_steps=100
)

print(f"VRMSE: {result.vrmse:.4f} | Latency: {result.latency_ms}ms")
result.export("./output_rollout.hdf5")

🏗️ Model Zoo & Benchmark Baselines

Model Architecture Type Top Domains Typical Latency (p95)
CNextU-Net Modernized ConvNeXt U-Net Turbulence, Compressible Euler, Active Matter 47ms
FNO Fourier Neural Operator Helmholtz Scattering, MHD, Thermal Convection 31ms
TFNO Tucker-Factorized FNO Shear Flow, Rayleigh-Bénard Convection 28ms
AViT Attention Vision Transformer Supernova Explosions, Neutron Star Mergers 89ms
AFNO Adaptive Fourier Neural Operator Viscoelastic Instabilities, Convective Envelopes 35ms
DilatedResNet Dilated Residual Network Acoustic Scattering, Gray-Scott Reaction-Diffusion 22ms

🌐 Open-Core vs. Kontinua Cloud

Feature Open-Source Core (kontinua) Kontinua Cloud (Managed Platform)
License Apache 2.0 (Free Forever) Commercial SaaS & Enterprise
Compute Execution Local CPU / Self-Hosted GPU Auto-Scaling H100/L4 Triton Cluster
Pre-Trained 2D Weights ✅ Included (Hugging Face) ✅ Included (Auto-Updated)
High-Res 3D Foundation Models ❌ (Manual Training) ✅ Pre-trained & Served via API
Managed Fine-Tuning ❌ (Run your own cluster) ✅ 1-Click Fine-Tuning Pipeline
Inference Latency Hardware dependent Sub-50ms p95 global SLA
Private Benchmark CI/CD ❌ ✅ Automated regression testing
Team RBAC & SOC 2 Compliance ❌ ✅ Enterprise Ready

🔬 Dataset & Scientific Citation

Kontinua is built on the 15TB The Well benchmark published at NeurIPS 2024 by Polymathic AI in collaboration with the Flatiron Institute, Cambridge, NYU, Princeton, UC Berkeley, Los Alamos, and Cornell.

@article{ohana2024well,
  title={The Well: A Large-Scale Collection of Diverse Physics Simulations for Machine Learning},
  author={Ohana, Ruben and McCabe, Michael and Meyer, Lucas and Morel, Rudy and Agocs, Fruzsina and Beneitez, Miguel and Berger, Marsha and Burkhart, Blakesly and Dalziel, Stuart and Fielding, Drummond and others},
  journal={Advances in Neural Information Processing Systems (NeurIPS)},
  volume={37},
  pages={44989--45037},
  year={2024}
}

📄 License & Community

Metadata

Release files for kontinua 1.2.1

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

Source distribution (sdist)

Source distribution for kontinua 1.2.1
File Size Uploaded
kontinua-1.2.1.tar.gz 86.8 kB Details

Built distribution (wheel)

Table of built distributions (wheels) for kontinua 1.2.1
File Interpreter ABI Platform
kontinua-1.2.1-py3-none-any.whl Python 3 none any Details

Total release size: 180.0 kB

Release files / kontinua-1.2.1.tar.gz

Download URL kontinua-1.2.1.tar.gz
Size 86.8 kB
Tags Source
SHA-256 checksum
How to use checksums
5664e6097a31e5c02d324d22193b5ededf432bbc3adac45b0b66fa9aa899dd05
BLAKE2b-256 checksum
How to use checksums
eba0dd758d4e4ff80434214106e95683be369f8282401c094176aa46465d41b8
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/7.0.0 CPython/3.13.7

Release files / kontinua-1.2.1-py3-none-any.whl

Download URL kontinua-1.2.1-py3-none-any.whl
Size 93.2 kB
Tags Python 3
SHA-256 checksum
How to use checksums
7c65419d2157fcbe6a27aa9efdf37a0f6f76ee2b6d59bd42200211e3ae926d31
BLAKE2b-256 checksum
How to use checksums
8dcb412f1d7573abc20e52feacd9f9e8315da10c456e144b554bc77190f19a76
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/7.0.0 CPython/3.13.7

Release history Release notifications | RSS feed

This release

1.2.1 This release

2 release files

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