-
Notifications
You must be signed in to change notification settings - Fork 200
Insights: apple/axlearn
Overview
-
0 Active issues
-
- 11 Merged pull requests
- 5 Open pull requests
- 0 Closed issues
- 0 New issues
Loading
Could not load contribution data
Please try again later
Loading
11 Pull requests merged by 7 people
-
Adds
convert_dense_to_moe_parameters
.#584 merged
Jul 21, 2024 -
Only donate
prebuilt_model_state
toinit_computation
.#599 merged
Jul 21, 2024 -
An implementation of pipeline schedules and StreamSchedule.
#598 merged
Jul 19, 2024 -
Enable FlashAttention Falling Back to Plain MHA for Decoding
#594 merged
Jul 18, 2024 -
Minor changes.
#593 merged
Jul 17, 2024 -
Add Dataflow Inference Examples
#470 merged
Jul 17, 2024 -
Adds support for hybrid mesh in aot compilation.
#590 merged
Jul 17, 2024 -
Makes trainer.py donate both prng_key and prebuilt_model_state for
init_computation
.#589 merged
Jul 17, 2024 -
[5/N] Add evaluator for open_api inference
#588 merged
Jul 16, 2024 -
BaseBastionManagedJob should use UTC timezone consistently
#587 merged
Jul 16, 2024 -
Add
SpmdTrainer.Config.context_manager
.#585 merged
Jul 15, 2024
5 Pull requests opened by 5 people
-
Make the ASR Encoder also pass back Spectrogram
#591 opened
Jul 17, 2024 -
Add job_id field in BastionManagedJob
#592 opened
Jul 17, 2024 -
Changes `init_computation` in trainer.py not to take prebuilt params.
#595 opened
Jul 18, 2024 -
Allow an AXLearn module to be used as an `optimizer` in `Learner`.
#596 opened
Jul 18, 2024 -
[research] add mmau readme
#597 opened
Jul 18, 2024
2 Unresolved conversations
Sometimes conversations happen on old items that aren’t yet closed. Here is a list of all the Issues and Pull Requests with unresolved conversations.
-
Add bastion_tier label in bastion code
#544 commented on
Jul 19, 2024 • 2 new comments -
Gradient Accumulation in Axlearn
#465 commented on
Jul 17, 2024 • 1 new comment