Skip to content

Pull requests: apple/axlearn

New pull request New
Clear current search query, filters, and sorts
45 Open 870 Closed
45 Open 870 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 back job priority.
#86 by markblee was merged Sep 28, 2023 Loading…
Pins pyOpenSSL.
#100 by markblee was merged Oct 6, 2023 Loading…
Support remat and its HF weights conversion for Bert-based text embedding encoder
#99 by JinhaoLei was merged Oct 5, 2023 Loading…
@markblee
Add content hinting for summary logging
#98 by apghml was merged Oct 5, 2023 Loading…
Cleanup tempdirs in summary writer tests.
#97 by markblee was merged Oct 4, 2023 Loading…
allow parse_kv_flags to accept multiple delimiter
#96 by zhiyun was merged Oct 4, 2023 Loading…
Add @no_remat decorator
#94 by apghml was merged Oct 4, 2023 Loading…
Initial documentation and AXLearn tutorial.
#92 by markblee was merged Oct 3, 2023 Loading…
Changes the API of input_tf_data.tfds_dataset
#91 by ruomingp was merged Oct 4, 2023 Loading…
Adds ResNet ImageNet trainer.
#90 by markblee was merged Oct 1, 2023 Loading…
Adds support for torch param conversion for LayerNormStateless.
#89 by ruomingp was merged Oct 1, 2023 Loading…
4
enable argnum donation on all architectures, call _step_log() for early steps, and add JAX scope annotations
#88 by apghml was merged Oct 3, 2023 Loading…
have inference runner use the same custom create_device_mesh as trainer
#87 by JinhaoLei was merged Sep 29, 2023 Loading…
@markblee
reduce logging
#73 by pdufter was merged Sep 15, 2023 Loading…
Makes Initializer a subclass of Configurable for consistency.
#85 by ruomingp was merged Sep 28, 2023 Loading…
Make tracebacks in Module easier to read
#84 by apghml was merged Oct 3, 2023 Loading…
Use hyperlinkable output format for Pylint
#83 by apghml was merged Sep 28, 2023 Loading…
Use BertLMHead as vocab_mapping child of SpladePooling layer
#82 by JinhaoLei was merged Sep 27, 2023 Loading…
@markblee
Improves tpu runner status check and test coverage.
#81 by markblee was merged Sep 26, 2023 Loading…
Modify trainer's _create_device_mesh method for better behavior on GPU.
#80 by tgunter was merged Sep 24, 2023 Loading…
Fixes some bugs with bastion user state handling.
#79 by markblee was merged Sep 25, 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…
Increase sharding-spec flexibility in trainer.py and attention.py
#75 by tgunter was merged Sep 18, 2023 Loading…
Adds a generic launch job.
#74 by markblee was merged Sep 18, 2023 Loading…
ProTip! Adding no:label will show everything without a label.