Welcome to JAX-AM’s documentation!#
JAX-AM is a Python library for numerical simulations in additive manufacturing.
Note
This project is under active development.
Scientific Simulations with JAX#
We use JAX for implementation of the computationally intensive part.
Why JAX? We believe JAX is an exciting tool for scientific simulations.
The code is in Python, and it runs on CPU/GPU with high performance.
It is natural for research of AI for science/engineering, because JAX is intended for machine learning research.
The automatic differentiation feature of JAX is useful for design and optimization.