JAX-FEM: A differentiable GPU-accelerated 3D finite element solver for automatic inverse design and mechanistic data science
Tianju Xue, Shuheng Liao, Zhengtao Gan, Chanwook Park, Xiaoyu Xie, Wing Kam Liu, Jian Cao
Topics & Concepts
Finite element methodSolverComputer sciencePython (programming language)ComputationComputational scienceDifferentiable functionApplied mathematicsNonlinear systemAlgorithmStructural engineeringProgramming languageMathematicsEngineeringMathematical analysisPhysicsQuantum mechanicsTopology Optimization in EngineeringComposite Material MechanicsModel Reduction and Neural Networks