Skip to content

Add CLAUDE.md documenting architecture, decorator system, and JAX rules#392

Merged
Jammy2211 merged 1 commit intomainfrom
feature/jax_decorator_bypass
Mar 6, 2026
Merged

Add CLAUDE.md documenting architecture, decorator system, and JAX rules#392
Jammy2211 merged 1 commit intomainfrom
feature/jax_decorator_bypass

Conversation

@Jammy2211
Copy link
Owner

Summary

  • Adds CLAUDE.md documenting PyAutoLens architecture, dependency chain, key directories, and how the @to_array / @to_grid / @to_vector_yx decorators are inherited from PyAutoArray
  • Documents the JAX jit boundary constraint and the if xp is np: guard pattern used in autogalaxy/operate/lens_calc.py

Key content

  • Architecture: Tracer, dataset/fit/analysis class table, key directory map
  • Decorator system: output-type-by-grid-type table, stacking order, .array access pattern
  • JAX rules: why autoarray types can't be JIT outputs, which LensCalc methods implement the guard

🤖 Generated with Claude Code

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
@Jammy2211 Jammy2211 merged commit 3582e18 into main Mar 6, 2026
4 of 8 checks passed
@Jammy2211 Jammy2211 deleted the feature/jax_decorator_bypass branch March 6, 2026 16:18
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant