Tools & Frameworks

Solvers, training frameworks, post-processing & visualization tools

Filter by tag:

20 resources

ClimateLearn: Benchmarking Machine Learning for Weather and Climate

Climate ML Benchmarking Library
ClimateLearn (UCLA Machine Intelligence Group, 2023) is a Python library that provides standardized access to climate datasets, baseline models, and metrics for large-scale benchmarking of statistical downscaling and temporal forecasting methods. Code is MIT licensed.
New MIT ClimateMachine LearningBenchmark

GROMACS

Biomolecular Molecular Dynamics
A high-performance molecular dynamics package focused on biomolecules (proteins, lipids, nucleic acids), renowned for extreme parallelism and vectorization — a mainstay of structural biology and drug design.
New GNU Lesser General Public License v2.1 Molecular DynamicsBiomolecularProtein

JAX

Differentiable Numerical Computing Framework
Google's open-source NumPy-style numerical computing library built on XLA with autodiff and functional transformations (grad/jit/pmap/vmap) — the substrate behind physics-AI differentiable solvers and neural operators like JAX-CFD and JAX-Fluids.
New Apache License 2.0 Automatic DifferentiationXLANumerical Computing

LAMMPS

Molecular Dynamics & Many-Body Simulation
A Sandia-led open-source molecular dynamics and particle simulator supporting multi-scale models from atomic to coarse-grained and continuum, with a large library of force fields — the workhorse for materials, chemistry and biomolecular simulation.
New GNU General Public License v2.0 Molecular DynamicsMany-Body SimulationForce Field

NVIDIA Warp

Differentiable GPU Simulation Framework
NVIDIA's open-source Python framework that JIT-compiles plain Python functions into CPU/CUDA kernels with automatic differentiable adjoints, letting gradients flow through entire physics simulations.
New Apache License 2.0 Differentiable SimulationGPUJIT Compilation

PyTorch Geometric (PyG)

Graph Neural Network Framework
A PyTorch-based library for graph neural networks offering rich GNN operators, popular datasets and samplers — the de facto standard for graph representation learning on molecules, crystals and physical systems.
New MIT License Graph Neural NetworkGNNPyTorch

Taichi Lang

Physics Simulation DSL
A Python-embedded DSL for high-performance computing that turns Python into parallel, auto-differentiable kernels, with built-in MPM/SPH/FEM frontends for physical simulation.
New Apache License 2.0 Physical SimulationDifferentiable ProgrammingDSL

CompFlowLab: A Python Environment for Developing Data-Driven Compressible Flow Models

CFD
CompFlowLab is an open-source Python environment for data-driven modeling of compressible flow problems, including shocks, flames, and detonation waves. It combines a one-dimensional compressible Navier-Stokes solver with multispecies transport and chemical-reaction models for lightweight prototyping and evaluation.
See LICENSE file Compressible FlowData-Driven ModelingChemical Reaction Flows

PDE-OBS Partial Differential Equation Model

ScientificML
This Hugging Face repository hosts the pdeobs-model resource under PDE-OBS, targeting scientific machine-learning workflows based on partial differential equations. The source listing provides no details about the model architecture, training data, benchmarks, or reported performance.
See LICENSE file Partial Differential EquationsScientific Machine LearningModel Resource

AstroGenesis: A Domain-Specific Multi-Agent AI for Astrophysical Research

Astrophysics
A domain-specific multi-agent AI framework for astrophysical research that combines literature retrieval, multiwavelength data access and analysis, theoretical modeling, and research ideation within a unified environment, with its current implementation focused on blazar research.
See LICENSE file AstrophysicsMulti-Agent AIBlazar Research

InternS1-Mini Context-Conditioned Molecule Transfer Model

Quantum/Chemistry
A context-conditioned molecule transfer model based on InternS1-Mini, designed for drug-induced liver injury (DILI) related molecular transformation tasks in chemical reaction prediction and drug discovery.
See LICENSE file molecule transferdrug discoverychemical reaction prediction

Context-Conditioned Molecule Transfer Model for Carcinogens

ScientificML
A deep learning-based context-conditioned molecule transfer model designed for carcinogen-related molecular structure migration and generation in cheminformatics.
See LICENSE file molecule transferchemistry discoverymolecular generation

NVIDIA Earth-2: AI-Powered Weather Decision Platform

CFD
NVIDIA Earth-2 leverages AI atmospheric models integrated with observation data to provide high-resolution weather forecasts and decision support for weather-sensitive industries.
See LICENSE file weather forecastingatmospheric modelingAI meteorology

Unitree UnifoLM-WLA-1.0 General-Purpose Humanoid Robot Foundation Model

Embodied AI / Robot Foundation Model
Unitree releases UnifoLM-WLA-1.0, a 6B parameter general-purpose embodied AI model trained with ~2500 hours of real-machine data, achieving 7 open-source SOTA results in embodied reasoning.
See LICENSE file embodied AIhumanoid robotfoundation model

PINN Heat Equation Solver

PINN
A heat conduction equation solver based on Physics-Informed Neural Networks, using neural networks embedded with heat conduction physics to predict temperature fields.
See LICENSE file Physics-Informed Neural NetworksHeat EquationHeat Conduction

SPUS-PDE-UNet-36M: Neural PDE Solver with 36M Parameters

ScientificML
A 36M-parameter neural network solver for partial differential equations based on U-Net architecture, enabling data-driven solving and prediction of PDE systems
See LICENSE file PDE solverneural networkU-Net
1–20 / 48
1 2 3
Go to