# Architecture

::::{grid} 1 2 2 2
:gutter: 2

:::{grid-item-card} 🗺️ Overview
:link: architecture/architecture_overview
:link-type: doc

High-level overview of MaxText design and components.
:::

:::{grid-item-card} 📚 JAX/AI Libraries
:link: architecture/jax_ai_libraries_chosen
:link-type: doc

Deep dive into the JAX and AI libraries used in MaxText.
:::
::::

```{toctree}
---
hidden:
maxdepth: 1
---
architecture/architecture_overview.md
architecture/jax_ai_libraries_chosen.md
```