AGENTS.md@src/gnn/analysis/jax git:20260906.e62b9e9A

Immutable. This exact content is served forever at /api/v1/blob/f6f193c5287f1471.

# JAX Analysis - Agent Scaffolding

## Overview

Framework-specific analyzer for JAX simulation results. Part of the Analysis module (Step 16).

## Module Structure

```
analysis/jax/
├── __init__.py    # Public API
├── analyzer.py    # JAXAnalyzer class
├── README.md      # Human documentation
└── AGENTS.md      # This file
```

## Key Functions

### analyzer.py

- `generate_analysis_from_logs(execution_dir, output_dir, verbose)` - Main entry point
- `_parse_jax_outputs(filepath)` - Parse JAX simulation outputs
- `_analyze_gradients(data)` - Gradient analysis
- `_analyze_convergence(data)` - Convergence tracking
- `_generate_report(metrics)` - Report generation

## Integration Points

**Upstream:** Execute module (Step 12) produces JAX simulation results
**Downstream:** Report module (Step 23) consumes analysis outputs

## Dependencies

- numpy: Numerical operations
- jax (optional): Native JAX analysis
- matplotlib (optional): Visualization

---

**Version:** 3.0.0
**Last Updated:** 2026-01-23