# API Reference

Complete API documentation for all JaxARC modules, auto-generated from source
code docstrings.

JaxARC provides a JAX-native environment for training agents on ARC (Abstraction
and Reasoning Corpus) tasks. The API is organized into several key modules:

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

:::{grid-item-card} Core API :link: core :link-type: doc

`make()`, `Environment`, `State`, `Action` - Essential functions and classes for
creating and running environments :::

:::{grid-item-card} Registration :link: registration :link-type: doc

Environment registration system, task IDs, and dataset management :::

:::{grid-item-card} Wrappers :link: wrappers :link-type: doc

Action and observation wrappers for transforming environment interface :::

:::{grid-item-card} Configuration :link: config :link-type: doc

`JaxArcConfig` and configuration system with Hydra integration :::

:::{grid-item-card} Types :link: types :link-type: doc

Type definitions: `EnvParams`, `TimeStep`, and other core types :::

:::{grid-item-card} Utilities :link: utils :link-type: doc

Helper functions for visualization, data loading, and more :::

::::

## Module Index

```{toctree}
:maxdepth: 2

core
registration
wrappers
config
types
utils
```
