FEAX is a compact, high-performance finite element analysis engine built on JAX. It provides an API for solving partial differential equations with automatic differentiation, JIT compilation, and GPU acceleration.
2 comments
Show HN: Finite Element Analysis with JAX | Heykuki News