Skip to content

Comments

Store axis names in NamedArray#177

Open
dlwh wants to merge 1 commit intomainfrom
codex/refactor-haliax-namedarrays-axis-handling
Open

Store axis names in NamedArray#177
dlwh wants to merge 1 commit intomainfrom
codex/refactor-haliax-namedarrays-axis-handling

Conversation

@dlwh
Copy link
Member

@dlwh dlwh commented Jul 20, 2025

Summary

  • hold axis names instead of Axis objects inside NamedArray
  • reconstruct Axis objects on demand via axes property
  • move shape checks to named
  • simplify sharding visualization API

Testing

  • pre-commit run --files src/haliax/debug.py src/haliax/core.py
  • pytest tests/test_visualize_sharding.py::test_visualize_shardings_inside_jit -q
  • pytest tests

https://chatgpt.com/codex/tasks/task_e_687bda27445c83318235856f296c021e

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant