Flight Log / stanford-crfm/levanter #1058
Update sharding helper to avoid deprecated JAX API — a OpenAI Codex (OpenAI) flight by David Hall
David Hall (@dlwh) opened this OpenAI Codex (OpenAI) pull request in stanford-crfm/levanter on Jul 13, 2025; merged clean in 49 min, merged by @dlwh: +14 −11, 3 files, 0 reviews.
Flown by
- Lines
- +14 −11
- Files
- 3
- Commits
- 1
- Reviews
- 0
- To merge
- 49 min
- Task
- fix
Line and file counts are summed per commit from the AIDev dataset; GitHub's net diff can be smaller.
OpenAI Codex, in stanford-crfm/levanter, fixes the sharding helper to stop using a deprecated JAX API. It switches best_effort_sharding to NamedSharding and drops PositionalSharding across three files, with 14 additions and 11 deletions in one commit. The pytest run fails on a missing ray module. The PR merged clean after 0.82 hours, with no changes requested.
From the pull request description
## Summary - switch `best_effort_sharding` to build a `NamedSharding` - fix imports in tests and stop using `PositionalSharding` ## Testing - `pre-commit run --all-files` - `pytest tests -m "not entry and not slow and not ray"` *(fails: ModuleNotFoundError: No module named 'ray')* ------ https://c…
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 13, 2025labeleddlwh
- Jul 13, 2025mergeddlwh
- Jul 13, 2025closeddlwh
Read next
Cite this flight: AgentRiders Flight Log, stanford-crfm/levanter#1058, record updated Nov 4, 2026, https://agentriders.com/logs/stanford-crfm/levanter/1058/
Data from public GitHub activity; this page is not affiliated with GitHub or the repository.