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:

Core API

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

Core API

Registration

Environment registration system, task IDs, and dataset management

Registration API

Wrappers

Action and observation wrappers for transforming environment interface

Wrappers API

Configuration

JaxArcConfig and configuration system with Hydra integration

Configuration API

Types

Type definitions: EnvParams, TimeStep, and other core types

Types API

Utilities

Helper functions for visualization, data loading, and more

Utilities API

Module Index

• Core API
  • Quick Example
  • Environment Creation
  • Core Classes
• Registration API
  • Module Contents
  • Task ID Format
• Wrappers API
  • Action Wrappers
  • Observation Wrappers
  • Visualization Wrappers
  • Usage Example
  • See Also
• Configuration API
  • Module Contents
  • Usage Examples
• Types API
  • Module Contents
  • Usage Example
  • JAX Typing
• Utilities API
  • Visualization
  • Core Utilities
  • Usage Examples
  • See Also