Skip to content

Pull requests: apple/axlearn

New pull request New
Clear current search query, filters, and sorts
32 Open 129 Closed
32 Open 129 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
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Add litepod config for 70B model
#656 by kelvin-zou was closed Sep 5, 2024 • Draft
6
Reduces shuffle buffer size for pajama_trainer.py.
#648 by ruomingp was closed Aug 12, 2024 • Draft
Use paddings parameter in conv_norm
#646 by stefbraun was closed Aug 9, 2024 Loading…
Conformer: use paddings arg in conv_norm layer
#644 by stefbraun was closed Aug 9, 2024 Loading…
Set remat_spec.policy None for fuji v2 70B
#638 by samos123 was closed Aug 8, 2024 Loading…
7
Add Orbax Checkpointing
#627 by jiya-zhang was closed Aug 21, 2024 • Draft
13
Change the CloudBuild to Sync by Default
#616 by amcw7777 was closed Jul 31, 2024 Loading…
Add option to return module outputs from inference runner
#611 by a-metz was closed Jul 26, 2024 Loading…
Add option to return module outputs from inference runner
#610 by a-metz was closed Jul 26, 2024 Loading…
Add option to return module outputs from inference runner
#609 by a-metz was closed Jul 26, 2024 Loading…
BaseBastionManagedJob should use UTC timezone consistently
#586 by HaijingFu was closed Jul 15, 2024 Loading…
Adds convert_dense_to_moe_parameters .
#584 by ruomingp was merged Jul 21, 2024 Loading…
1
New DataPartitionType DATA
#567 by apoorvtintin was closed Dec 11, 2024 Loading…
6
Neuron support in Axlearn
#566 by apoorvtintin was closed Jan 12, 2025 Loading…
16
add kv_state arg to forward function in RoFormerQKVLinear
#556 by chongw was closed Jun 28, 2024 Loading…
3
Test pr push
#553 by zhiyun was closed Jun 27, 2024 Loading…
Test pr push
#552 by zhiyun was closed Jun 27, 2024 Loading…
Test pr push
#551 by zhiyun was closed Jun 27, 2024 Loading…
Test pr push
#550 by zhiyun was closed Jun 27, 2024 Loading…
Test pr push
#549 by zhiyun was closed Jun 27, 2024 Loading…
Fix codeowners.
#548 by markblee was merged Jun 27, 2024 Loading…
1
Add github action support
#542 by samos123 was closed Jan 9, 2025 Loading…
4
Request memory for TPUGKEJob pod based on TPU machine type
#525 by HaijingFu was closed Jun 18, 2024 Loading…
2
Enable GKE node pool pre-provisioning
#514 by Ethanlm was closed Jun 7, 2024 Loading…
Add SummaryWriter log_checkpoint to Checkpointer as optional.
#490 by garyshincc was closed May 24, 2024 Loading…
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.