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

Fixes lion optimizer dtypes.
#36 by markblee was merged Aug 17, 2023 Loading…
Removes _auto_child_context_method_names.
#39 by markblee was merged Aug 22, 2023 Loading…
Removes unused DummyDecodingModel.
#46 by markblee was merged Aug 25, 2023 Loading…
Adds support for generics in configs.
#44 by markblee was merged Aug 26, 2023 Loading…
Adds the axlearn.cloud module.
#40 by markblee was merged Aug 22, 2023 Loading…
Adds support for beam search prefilling.
#42 by markblee was merged Aug 24, 2023 Loading…
Supports configuring causal=True optimization in FlashAttention mha.
#31 by markblee was merged Aug 7, 2023 Loading…
Adds axlearn gcp cli.
#54 by markblee was merged Sep 1, 2023 Loading…
Supports retaining intermediate output collections across scan.
#56 by markblee was merged Sep 2, 2023 Loading…
Supports stack-of-stacks and repeat-of-stacks in transformer decoding.
#58 by markblee was merged Sep 6, 2023 Loading…
Support chex.dataclass in tree_paths.
#22 by markblee was merged Aug 3, 2023 Loading…
Adds a flash attention layer.
#26 by markblee was merged Aug 4, 2023 Loading…
Allow alpha=0 in lora.
#28 by JianyuWangV was merged Aug 7, 2023 Loading…
3
Add @no_remat decorator
#95 by apghml was closed Oct 3, 2023 Loading…
Cleanup tempdirs in summary writer tests.
#97 by markblee was merged Oct 4, 2023 Loading…
Supports configuring instances as dst_layer in hf builder.
#71 by markblee was merged Sep 14, 2023 Loading…
Supports regex patterns in quota file.
#77 by markblee was merged Sep 21, 2023 Loading…
Makes vertexai imports optional.
#76 by markblee was merged Sep 19, 2023 Loading…
Introduces SpmdTrainer._train_step_input_partition_specs()
#202 by ruomingp was merged Nov 25, 2023 Loading…
Relax jax version in dependencies.
#205 by markblee was closed Nov 27, 2023 Loading…
1
update dataflow launch script in docstring
#196 by zhiyun was closed Dec 5, 2023 Loading…
Approval for CI
#225 by madrob was merged Dec 5, 2023 Loading…
Update config.yml [DO NOT MERGE]
#226 by madrob was closed Dec 5, 2023 Loading…
Lazy import tfio.
#233 by markblee was merged Dec 7, 2023 Loading…
Add Request ID to HTTP requests
#35 by jiya-zhang was closed Dec 14, 2023 Loading…
4
ProTip! Exclude everything labeled bug with -label:bug.