GitHub repository: tianjuxue/jax-fem
Primary language: Python
Description: Differentiable Finite Element Method with JAX