Skip to yearly menu bar Skip to main content

Workshop: Differentiable Almost Everything: Differentiable Relaxations, Algorithms, Operators, and Simulators

JAX FDM: A differentiable solver for inverse form-finding

Rafael Pastrana · Deniz Oktay · Ryan P. Adams · Sigrid Adriaenssens


We introduce JAX FDM, a differentiable solver to design mechanically efficient shapes for 3D structures, such as domes, cable nets and towers, conditioned on target architectural, fabrication and structural properties. JAX FDM solves these inverse form-finding problems by combining the force density method, differentiable sparsity and gradient-based optimization. JAX FDM can be paired with optimization and neural network libraries in the JAX ecosystem to facilitate the integration of form-finding simulations into neural networks. We showcase the features of JAX FDM in two structural design examples. JAX FDM is available as an open-source library.

Chat is not available.