Skip to content

Pull requests: marin-community/haliax

New pull request New
15 Open 164 Closed
15 Open 164 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Use stable jax.shard_map import path
#229 opened Jun 23, 2026 by rfl-urbaniak Loading…
2
loose end: Partial vmap
#201 opened Aug 24, 2025 by dlwh Member Loading…
loose-end (destructure unordered axis)
#200 opened Aug 24, 2025 by dlwh Member Loading…
test: duplicate axis mapping raises error codex
#197 opened Aug 19, 2025 by dlwh Member Loading…
2
Implement shard_map wrapper codex
#188 opened Jul 29, 2025 by dlwh Member Loading…
Use shard in named_jit codex
#178 opened Jul 21, 2025 by dlwh Member Loading…
Store axis names in NamedArray codex
#177 opened Jul 20, 2025 by dlwh Member Loading…
Add generic axis resolution in check_axes codex
#173 opened Jul 15, 2025 by dlwh Member Loading…
Update docs to clarify axis usage codex
#152 opened Jul 5, 2025 by dlwh Member Loading…
Add synthetic MNIST data example codex
#150 opened Jul 5, 2025 by dlwh Member Loading…
Add tests for multi-generic axes codex
#146 opened Jun 29, 2025 by dlwh Member Loading…
feat: Add CNN example validated by executed script
#144 opened Jun 28, 2025 by dlwh Member Loading…
tweaks/bugfixes
#137 opened Jun 24, 2025 by dlwh Member Loading…
improve einsum/rearrange parsing
#129 opened Mar 28, 2025 by dlwh Member Loading…
Mixed Precision and Resource Envs
#65 opened Feb 6, 2024 by dlwh Member Loading…
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.