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.

Solving Problems in Additive Manufacturing

Temperature field

Melt pool

Microstructure

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.

1. The code is in Python, and it runs on CPU/GPU with high performance.

2. It is natural for research of AI for science/engineering, because JAX is intended for machine learning research.

3. The automatic differentiation feature of JAX is useful for design and optimization.

Contents

Contents:

• Install
• Quick Start
• Notes
• Tutorials
• About

Indices and tables

• Index

• Module Index