Skip to content

Pull requests: apple/axlearn

New pull request New
Clear current search query, filters, and sorts
47 Open 843 Closed
47 Open 843 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

Adds the axlearn.cloud module.
#40 by markblee was merged Aug 22, 2023 Loading…
Fix flax loader
#24 by RileyLRoberts was merged Aug 3, 2023 Loading…
run pytest files in parallel on Circle CI
#25 by ya5ut was merged Aug 4, 2023 Loading…
2
Adds a flash attention layer.
#26 by markblee was merged Aug 4, 2023 Loading…
Enable building StackedTransformerLayer with a sequence of layer cfgs
#27 by xianzhidu was merged Aug 4, 2023 Loading…
1
Allow alpha=0 in lora.
#28 by JianyuWangV was merged Aug 7, 2023 Loading…
3
Add comments for the StackedTransformerLayer layer config.
#29 by xianzhidu was merged Aug 6, 2023 Loading…
[gcp]: migrate google oauth2 credentials to google auth credentials
#43 by zhzhyi was merged Aug 23, 2023 Loading…
Adds initial GCP utilities.
#33 by markblee was merged Aug 8, 2023 Loading…
Revert "fix l2_norm (#21)"
#34 by markblee was merged Aug 10, 2023 Loading…
Add Request ID to HTTP requests
#35 by jiya-zhang was closed Dec 14, 2023 Loading…
4
Fixes lion optimizer dtypes.
#36 by markblee was merged Aug 17, 2023 Loading…
Supports prefix padding with sample_decode prefill.
#38 by markblee was merged Aug 20, 2023 Loading…
Remove hf_pretrained_loaders.py.
#12 by tuzhucheng was merged Jul 26, 2023 Loading…
@markblee
Make logits partition specs configurable in decoder.
#41 by xianzhidu was merged Aug 23, 2023 Loading…
2
Adds support for beam search prefilling.
#42 by markblee was merged Aug 24, 2023 Loading…
Provides a brief overview of AXLearn in README.md
#30 by ruomingp was merged Aug 6, 2023 Loading…
Add l2 norms for the VQ-VAE quantizer.
#2 by xianzhidu was merged Jul 18, 2023 Loading…
Quick fix
#3 by xianzhidu was merged Jul 19, 2023 Loading…
expose concurrent_gb in state builder
#4 by gyin94 was merged Jul 21, 2023 Loading…
Fix some permlinks
#5 by xianzhidu was merged Jul 22, 2023 Loading…
1
Fix typo: os.kath -> os.path
#6 by llcao was merged Jul 20, 2023 Loading…
relax learner state to be a general dictionary
#8 by zhiyun was merged Jul 23, 2023 Loading…
Update README.md
#9 by TanayShukla was closed Aug 5, 2023 Loading…
Fix L2 Norm
#21 by gokul-uf was merged Aug 8, 2023 Loading…
ProTip! Updated in the last three days: updated:>2025-01-15.