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, autodiff-native software components for Simulation Intelligence. 📦

phaethon

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

diffeqpy

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

opinf

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

jax-neural-operators

JAX library for neural operator and PDE foundation model training.

lirisimpact

Repository contaning the original code of IMPACT algorithm, an interpretable model for ordinal predictions with multi-class outputs"

verifiable-labs

RL environments for scientific AI agents with conformal-calibrated rewards

pyplaid

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

jwave

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

tesseract-jax

Execute + differentiate Tesseracts as part of JAX programs, with full support for function transformations like JIT, grad, and more. ⚡

tensordiffeq

Efficient and Scalable Physics-Informed Deep Learning and Scientific Machine Learning on top of Tensorflow for multi-worker distributed computing

rom-operator-inference

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

verifiable-labs-envs

RL environments for scientific AI agents with conformal-calibrated rewards

vlabs-calibrate

RL environments for scientific AI agents with conformal-calibrated rewards

reactorch

ReacTorch: A Differentiable Reacting Flow Simulation Package in PyTorch

unitflex

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

idrlnet

IDRLnet

modularml

A library for modular, fast, and reproducible ML experimentation built for R&D.

koopmanlab

A library for Koopman Neural Operator with Pytorch.

rtide

A Python implementation of the automated response method for tidal analysis and prediction of complex tidal phenomena.

reservoirflow

Reservoir Simulation and Engineering Library in Python

fastvpinns

A fast tensor-driven variational physics-informed neural network library for solving PDEs.