← All projects/ tensor-mesh.com/ Since Sept 2026

TensorMesh

A fast, differentiable, GPU-native finite element library for PyTorch — JIT-free, easy to debug, and end-to-end differentiable.

TensorMesh logo
/ About

TensorMesh is a fast, differentiable, GPU-native finite element library for PyTorch. It is JIT-free, easy to debug, and end-to-end differentiable, so finite element solvers can sit directly inside a training loop.

/ ETH AI Club Members involved
  • Shizheng WenShizheng Wen