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

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

kontinua-1.2.1.tar.gz (86.8 kB view details)

Uploaded Source

Built Distribution

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

kontinua-1.2.1-py3-none-any.whl (93.2 kB view details)

Uploaded Python 3

File details

Details for the file kontinua-1.2.1.tar.gz.

File metadata

  • Download URL: kontinua-1.2.1.tar.gz
  • Upload date:
  • Size: 86.8 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/7.0.0 CPython/3.13.7

File hashes

Hashes for kontinua-1.2.1.tar.gz
Algorithm Hash digest
SHA256 5664e6097a31e5c02d324d22193b5ededf432bbc3adac45b0b66fa9aa899dd05
MD5 a49c837bb761a728a4828fce65b58d0f
BLAKE2b-256 eba0dd758d4e4ff80434214106e95683be369f8282401c094176aa46465d41b8

See more details on using hashes here.

File details

Details for the file kontinua-1.2.1-py3-none-any.whl.

File metadata

  • Download URL: kontinua-1.2.1-py3-none-any.whl
  • Upload date:
  • Size: 93.2 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/7.0.0 CPython/3.13.7

File hashes

Hashes for kontinua-1.2.1-py3-none-any.whl
Algorithm Hash digest
SHA256 7c65419d2157fcbe6a27aa9efdf37a0f6f76ee2b6d59bd42200211e3ae926d31
MD5 4a517e4de770dbc26c82186c4bb04639
BLAKE2b-256 8dcb412f1d7573abc20e52feacd9f9e8315da10c456e144b554bc77190f19a76

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page