Scientific Machine Learning Python Packages

deepxde

A library for scientific machine learning and physics-informed learning

jaxns

Probabilistic Programming and Nested sampling in JAX

tesseract-core

Universal components for differentiable scientific computing. :package:

diffeqpy

Solving differential equations in Python using DifferentialEquations.jl and the SciML Scientific Machine Learning organization

adcd

Discover physical correction laws from anomalous data in JAX.

tesseract-jax

Tesseract JAX executes Tesseracts as part of JAX programs, with full support for function transformations like JIT, `grad`, and more.

opinf

Operator Inference for data-driven, non-intrusive model reduction of dynamical systems.

pyplaid

PLAID (Physics-Learning AI Datamodel), a flexible and extensible framework for representing and sharing datasets of physics simulations

lirisimpact

IMPACT framework, an interpretable multi-target framework for ordinal outputs

vlabs-calibrate

Open-source SDK (Apache-2.0): RL environments, conformal calibration, a TRL-compatible reward function, the Lean 4 formal track, and the verifiable/vlabs CLI.

jwave

A JAX-based research framework for differentiable and parallelizable acoustic simulations, on CPU, GPUs and TPUs

mosaic-bench

A benchmark suite for differentiable physics solvers

verifiable-labs-envs

Open-source SDK (Apache-2.0): RL environments, conformal calibration, a TRL-compatible reward function, the Lean 4 formal track, and the verifiable/vlabs CLI.

phaethon

End-to-End Physics-Constrained Scientific Computing & Sci-ML Framework

jax-neural-operators

JAX library for neural operator and PDE foundation model training.

tesseract-torch

Execute + differentiate Tesseracts as part of PyTorch programs, with full support for reverse-mode and forward-mode AD. 🔦

rom-operator-inference

Operator inference for data-driven, non-intrusive model reduction of dynamical systems.

tensordiffeq

Distributed PDE Solver in Tensorflow

reactorch

A Differentiable Reacting Flow Simulation Package in PyTorch

modularml

Modular, fast, and reproducible ML experimentation built for R&D.

unitflex

End-to-End Physics-Constrained Scientific Computing & Sci-ML Framework

idrlnet

IDRLnet

verifiable-labs

Python SDK + CLI for Verifiable Labs — evaluate frontier LLMs on conformal-calibrated scientific RL environments.

rtide

RTide: A python implementation of the ML Response Framework for Tidal Analyis and Prediction