Package: jax-am
A GPU-accelerated differentiable simulation toolbox for additive manufacturing (AM) based onJAX.JAX-AMJAX-AM is a collection of several numerical tools, currently includingDiscrete Element Method (DEM),Lattice Boltzmann Methods (LBM),Computational Fluid Dynamics (CFD),Phase Field Method (PFM)andFinite Element Method (FEM), that cover the analysis of theProcess-Structure-Propertyrelationship in AM.Our vision is to share with the AM community afree, open-source(under the GPL-3.0 License) software that facilitates the relevant computational research. In the JAX ecosystem, we hope to emphasize the potential of JAX for scientific computing. At the same time, AI-enabled research in AM can be made easy with JAX-AM.Authors:Tianju XueShuheng LiaoZhengtao Gan:fire:Join us for the development of JAX-AM!:rocket:Discrete Element Method (DEM)DEM simulation can be used for simulating powder dynamics in metal AM.Free falling of 64,000 spherical particles.Lattice Boltzmann Methods (LBM)LBM can simulate melt pool dynamics with a free-surface model.A powder bed fusion process considering surface tension, Marangoni effect and recoil pressure.Computational Fluid Dynamics (CFD)CFD helps to understand the AM process by solving the (incompressible) Navier-Stokes equations for velocity, pressure and temperature.Melt pool dynamics.Phase Field Method (PFM)PFM models the grain development that is critical to form the structure of the as-built sample.Microstructure evolution.Directional solidification with isotropic (left) and anisotropic (right) grain growth.Finite Element Method (FEM)FEM is a powerful tool for thermal-mechanical analysis in AM. We support the following features2D quadrilateral/triangle elements3D hexahedron/tetrahedron elementsFirst and second order elementsDirichlet/Neumann/Cauchy/periodic boundary conditionsLinear and nonlinear analysis includingHeat equationLinear elasticityHyperelasticityPlasticity (macro and crystal plasticity)Differentiable simulation for solving inverse/design problemswithouthuman deriving sensitivities, e.g.,Toplogy optimizationOptimal thermal controlThermal profile in direct energy deposition.Linear static analysis of a bracket.Crystal plasticity: grain structure (left) and stress-xx (right).Topology optimization with differentiable simulation.DocumentationPlease see theweb documentationfor the installation and use of this project.LicenseThis project is licensed under the GNU General Public License v3 - see theLICENSEfor details.CitationsIf you found this library useful in academic or industry work, we appreciate your support if you consider 1) starring the project on Github, and 2) citing relevant papers:@article{xue2023jax,title={JAX-FEM: A differentiable GPU-accelerated 3D finite element solver for automatic inverse design and mechanistic data science},author={Xue, Tianju and Liao, Shuheng and Gan, Zhengtao and Park, Chanwook and Xie, Xiaoyu and Liu, Wing Kam and Cao, Jian},journal={Computer Physics Communications},pages={108802},year={2023},publisher={Elsevier}}@article{xue2022physics,title={Physics-embedded graph network for accelerating phase-field simulation of microstructure evolution in additive manufacturing},author={Xue, Tianju and Gan, Zhengtao and Liao, Shuheng and Cao, Jian},journal={npj Computational Materials},volume={8},number={1},pages={201},year={2022},publisher={Nature Publishing Group UK London}}