About
Hi all! My name is Gaoyuan Wu (吴高远). I obtained my Ph.D. in civil engineering from Princeton University, advised by Prof. Maria Garlock. Prior to Princeton, I did my undergraduate studies in civil engineering at Tongji University with a focus in bridge engineering.
My research aims to promote coastal resilience of infrastructure from an interdisciplinary perspective that integrates structural engineering, architectural design, coastal engineering, optimization, and machine learning (ML). Specifically, my research uses numerical simulations, experiments, and Bayesian Optimization to investigate how shape optimization and modification of structures can lead to infrastructure that is more resilient and robust against extreme wave loads induced by climate-influenced hazards.
Besides, my research also involves developing differentiable finite element analysis (FEA) solver, JaxSSO, to enhance structural design via efficient gradient-based structural optimization - mitigating embodied carbon of the built environment while delivering structural elegance. I am also interested in training neural networks (NN) with FEA in the loop to benefit broad strutural design and optimization problems. For more, please see the Github repo: JaxSSO.
Outside research, I served as an Residential Graduate Student (RGS) at Forbes College for two years, from 2023 to 2025.
What’s New
- Our new work “An SPH study of cross-sectional shape effects on coastal structures subject to regular wave forces” and is available online. DOI: 10.1016/j.oceaneng.2025.122929. September, 2025
- Preprint “JAX-SSO: Differentiable Finite Element Analysis Solver for Structural Optimization and Seamless Integration with Neural Networks” is available online. arXiv link: arXiv:2407.20026. July 2024
- Our new work “Investigating the Effects of Box Girder Bridge Geometry on Solitary Wave Force Using SPH Modeling” was accepted by Coastal Engineering and is available online. DOI: 10.1007/s00158-023-03601-0. PDF is available here. November, 2023
- Our new work “A framework for structural shape optimization based on automatic differentiation, the adjoint method and accelerated linear algebra” was accepted by Structural and Multidisciplinary Optimization and is available online. DOI: 10.1007/s00158-023-03601-0. PDF is available here. Code for this project is available in its repo with a few examples, check them out. June, 2023
Projects
JaxSSO: A differentiable finite element analysis (FEA) solver for structural optimization, enabled by JAX.
Features
- Automatic differentiation (AD): an easy and accurate way for gradient evaluation. The implementation of AD avoids deriving derivatives manually or trauncation errors from numerical differentiation.
- Acclerated linear algebra (XLA) and just-in-time compilation: these features in JAX boost the gradient evaluation
- Hardware acceleration: run on GPUs and TPUs for faster experience
- Support beam-column elements and MITC-4 quadrilateral shell elements
- Shape optimization, size optimization and topology optimization
- Seamless integration with machine learning (ML) libraries
Here is an implementation of JaxSSO to form-find a structure inspired by Mannheim Multihalle using simple gradient descent. (First photo credit to Daniel Lukac)

FEM solver of structural dynamics problem for 2D frame system
This is my final project for APC523 (Numerical Algorithms for Scientific Computing). It can be used for the following problems:
- 2D-frame system under static loads.
- 2D-frame system under dynamic loads (such as earthquake motions)
A comparison between the solver and commercial software SAP2000 is shown below, in which the nodal displacement from the solver matches well with SAP2000: 
For more, please visit the GitHub repo.