Flight Log / stanford-crfm/haliax #178
Use shard in named_jit — a OpenAI Codex (OpenAI) flight by David Hall
David Hall (@dlwh) opened this OpenAI Codex (OpenAI) pull request in stanford-crfm/haliax on Jul 21, 2025; open and stale: +5 −11, 1 file, 0 reviews.
Flown by
- Lines
- +5 −11
- Files
- 1
- Commits
- 1
- Reviews
- 0
- To merge
- —
- Task
- refactor
Line and file counts are summed per commit from the AIDev dataset; GitHub's net diff can be smaller.
From the pull request description
## Summary - remove use of `out_shardings` in `named_jit` - instead apply `shard` inside the jitted function ## Testing - `uv run pre-commit run --all-files` - `XLA_FLAGS=--xla_force_host_platform_device_count=8 PYTHONPATH=tests:src:. uv run pytest tests` ------ https://chatgpt.com/codex/tasks/tas…
Excerpt (first 300 characters) — the full description is on GitHub.
View the pull request on GitHub →
This is your flight? Claim it — sign in with GitHub. Ownership is proven by the sign-in: the account that opened the PR is the only one that can claim it.
Flight timeline
- —committedDavid Hall
- Jul 21, 2025labeleddlwh
Read next
Cite this flight: AgentRiders Flight Log, stanford-crfm/haliax#178, record updated Nov 3, 2026, https://agentriders.com/logs/stanford-crfm/haliax/178/
Data from public GitHub activity; this page is not affiliated with GitHub or the repository.