Tools & Frameworks

Solvers, training frameworks, post-processing & visualization tools

AIPerf: LLM Inference Benchmarking at Scale

Benchmarking
AIPerf is an LLM inference benchmarking tool from NVIDIA designed to evaluate the speed and performance of large language models in deployment scenarios.
See LICENSE file benchmarkingLLM inferenceperformance evaluation AI Generated

HiDream-O1-Video by ZhiXiang

Video Generation/AI
HiDream-O1-Video-1.0 is a video generation model that demonstrates strong understanding of real-world scenarios, excelling in complex dynamic scenes like hip-hop rap and NBA game-winning shots, ranking among the top four globally.
See LICENSE file video generationAI videotext-to-video AI Generated

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 AI Generated

PhysicsNeMo v2.2.2

Physics Simulation
NVIDIA PhysicsNeMo v2.2.2 release fixes documentation alignment with the PyPi package name, improving installation accuracy.
Apache License 2.0 NVIDIAPhysics SimulationDeep Learning AI Generated

FEniCS / FFCx

Finite Element Framework / Multiphysics
Open-source FEM-based multiphysics simulation framework; describes variational problems in UFL and solves large-scale PDEs - a standard academic tool.
LGPL-3.0 FEMFinite ElementMultiphysics

JAX-Fluids

Differentiable CFD Solver
Fully auto-differentiable CFD solver on JAX (CPU/GPU/TPU) for end-to-end differentiable fluid simulation, coupling seamlessly with neural networks.
GNU General Public License v3.0 JAXDifferentiable CFDDifferentiable Physics

Modin

Data Analysis / Parallel Computing
Distributed DataFrame library, a drop-in pandas replacement that auto-parallelizes across cores/clusters (Dask/Ray/Intel) for large-scale simulation data analysis.
Apache-2.0 DataFrameParallelBig Data

PETSc

Numerical Solver Library / HPC
High-performance scientific linear/nonlinear solver library for large sparse systems - the underlying engine of top-tier CFD and multiphysics simulation.
BSD-2-Clause SolverLinear AlgebraSparse Matrix

PyVista

Visualization / Post-Processing
Python 3D visualization and mesh-processing framework for VTK - concise API for field data, mesh rendering and animation; the go-to viz library for simulation results.
MIT VisualizationVTKMesh

PhysicsNeMo

Solver / Training Framework
NVIDIA's physics-ML training framework (formerly Modulus) with PINN, neural operators and physics-constrained training - commercial-friendly, industry-first choice.
Apache-2.0 PINNNeural OperatorNVIDIA

FEniCS/dolfinx v0.11.0.post0 Released

Scientific Computing/Finite Element
FEniCS/dolfinx releases a patch version that fixes a Mathjax rendering issue in the built documentation. This release is functionally identical to v0.11.0, so users do not need to upgrade.
GNU Lesser General Public License v3.0 FEniCSdolfinxFinite Element AI Generated

DeepXDE

PINN Starter Framework
The most beginner-friendly PINN framework, with excellent Chinese docs and TensorFlow/PyTorch backends - the top choice to start with physics-informed networks.
GNU Lesser General Public License v2.1 PINNBeginnerChinese Docs

NeuralOperator

Neural Operator Library
Reference implementation library for FNO / DeepONet and other neural operators (MIT) - the de-facto standard for learning 'infinite-dimensional maps' as surrogates.
MIT Neural OperatorFNODeepONet

POSEIDON

PDE Foundation Model / Downstream Benchmark
ETH Zurich's PDE foundation model and downstream fine-tuning suite, using semigroup augmentation to mine abundant training pairs, covering 15 downstream PDE tasks.
See LICENSE file Foundation ModelNeural OperatorDownstream Tasks

OpenFOAM

General-Purpose CFD Solver
The de facto open-source CFD solver covering incompressible/compressible flow, heat transfer, multiphase and combustion - the workhorse for generating physics-AI training data.
GPL v3 CFDOpen SourceSolver

SU2

Aerodynamics / Design Optimization Solver
Open-source CFD and adjoint solver with built-in aerodynamic shape optimization, widely used for airfoil/aircraft analysis and gradient-driven design.
GNU Lesser General Public License v2.1 CFDAdjointShape Optimization

JAX-CFD

Differentiable Fluid Solver
Google Research's differentiable JAX-based fluid solver enabling end-to-end gradients for embedding neural networks into CFD time stepping.
Apache-2.0 JAXDifferentiable PhysicsCFD

DeepXDE v1.15.0

PDE Solving
DeepXDE v1.15.0 released with L-LAAF activation function support for JAX backend and Apple MPS GPU acceleration for PyTorch backend, improving deep learning-based PDE solving performance.
LGPL-2.1 PDEDeep LearningScientific Computing AI Generated

NeuralOperator v2.0.0 Release

Scientific Computing
NeuralOperator 2.0.0 major release introduces Codano model, improves documentation, and enhances usability for training Neural Operators in PyTorch.
MIT License Neural OperatorDeep LearningPyTorch AI Generated