Top AI Repos — open-source AI, indexed and scored
Top AI Repos tracks AI repositories on GitHub and answers two different questions about each one: is it moving right now, and would you bet a product on it.
Top AI Repos tracks AI repositories on GitHub and answers two different questions about each one: is it moving right now, and would you bet a product on it.
Additive manufacturing simulation with JAX.
| Date | Stars |
|---|---|
| 2026-07-25 | 360 |
| 2026-07-28 | 360 |
| 2026-07-30 | 360 |
| 2026-08-06 | 360 |
Today
— stars today
This week
— stars this week
This month
— stars this month
Momentum
0.0
growth rate 0.00%/day
<p align="middle">
<img src="docs/materials/logo.png" width="200" />
</p>
A GPU-accelerated differentiable simulation toolbox for additive manufacturing (AM) based on [JAX](https://github.com/google/jax).
# JAX-AM
[](https://jax-am.readthedocs.io/en/latest/)
JAX-AM is a collection of several numerical tools, currently including __Discrete Element Method (DEM)__, __Lattice Boltzmann Methods (LBM)__, __Computational Fluid Dynamics (CFD)__, __Phase Field Method (PFM)__ and __Finite Element Method (FEM)__, that cover the analysis of the __Process-Structure-Property__ relationship in AM.
Our vision is to share with the AM community a __free, 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.
:fire: ***Join us for the development of JAX-AM!***
## Discrete Element Method (DEM)
DEM simulation can be used for simulating powder dynamics in metal AM.
<p align="middle">
<img src="docs/materials/dem.gif" width="400" />
</p>
<p align="middle">
<em >Free falling of 64,000 spherical particles.</em>
</p>
## Lattice Boltzmann Methods (LBM)
LBM can simulate melt pool dynamics with a free-surface model.
<p align="middle">
<img src="docs/materials/lbm_pbf.gif" width="700" />
</p>
<p align="middle">
<em >A powder bed fusion process considering surface tension, Marangoni effect and recoil pressure.</em>
</p>
## Computational Fluid Dynamics (CFD)
CFD helps to understand the AM process by solving the (incompressible) Navier-Stokes equations for velocity, pressure and temperature.
<p align="middle">
<img src="docs/materials/melt_pool_dynamics.png" width="400" />
</p>
<p align="middle">
<em >Melt pool dynamics.</em>
</p>
## Phase Field Method (PFM)
PFM models the grain development that is critical to form the structure of the as-built sample.
<p align="middle">
<img src="docs/materials/single_track_eta_DNS.gif" width="600" />
</p>
<p align="middle">
<em >Microstructure evolution.</em>
</p>
<p align="middle">
<img src="docs/materials/solidification_isotropic.gif" width="300" />
<img src="docs/materials/solidification_anisotropic.gif" width="300" />
</p>
<p align="middle">
<em >Directional solidification with isotropic (left) and anisotropic (right) grain growth.</em>
</p>
## Finite Element Method (FEM)
:mega: :mega: :mega:
$${\color{red}We \space have \space decided \space to \space move \space the \space development \space of \space FEM \space to \space a \space separate \space repository.}$$
Check [JAX-FEM](https://github.com/tianjuxue/jax-fem). This is a design decision that aims to push the FEM module into a general-purpose, independent package that works for problems beyond additive manufacturing. Therefore, the code related to FEM in this repository (JAX-AM) will **NOT** be updated in the future.
## Documentation
Please see the [web documentation](https://jax-am.readthedocs.io/en/latest/) for the installation and use of this project.
## License
This project is licensed under the GNU General Public License v3 - see the [LICENSE](https://www.gnu.org/licenses/) for details.
## Citations
If 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:
```bibtex
@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, ZExcerpt of 4,582 characters
Read on GitHubWould you bet a product on this? Bounded 0–100 and slow moving.
matched fp:74734efcc340e037, topic:jax
matched fp:74734efcc340e037, topic:simulation