Skip to content

Activity

Update gke tpu job to bypass jobset coordinator

jesus-orozcopushed 1 commit to pathways_trillium • 9f19159…8f71508 • 
9 days ago

Update pathways container specs for gke tpu job

jesus-orozcopushed 1 commit to pathways_trillium • a0bf9df…9f19159 • 
9 days ago

Revert updates to fuji model config

jesus-orozcopushed 1 commit to pathways_trillium • c7bc5df…a0bf9df • 
11 days ago

remove unnecessary updates to gke tpu job for pathways workloads

jesus-orozcopushed 1 commit to pathways_trillium • ca5c883…c7bc5df • 
11 days ago

revert changes to trainer and model configs

jesus-orozcopushed 3 commits to pathways_trillium • d63dd6a…ca5c883 • 
11 days ago

refactor pathways jobset to new spec

jesus-orozcocreated backup_pathways_trillium • 89ac1ea • 
11 days ago

Rebase to axlearn main

jiya-zhangpushed 30 commits to pathways_trillium • 89ac1ea…d63dd6a • 
11 days ago

Introduce BaseAttentionBias.has_value(). (apple#920)

jiya-zhangpushed 9 commits to maggiejz/main • 60ca6ce…feb8357 • 
11 days ago

Implements FlashDecoding with Sparsity Support (apple#899)

jiya-zhangcreated maggiejz/main • 60ca6ce • 
11 days ago

refactor pathways jobset to new spec

jesus-orozcopushed 1 commit to pathways_trillium • 357a322…89ac1ea • 
14 days ago

Implements FlashDecoding with Sparsity Support (apple#899)

jesus-orozcopushed 12 commits to main • 73625c9…60ca6ce • 
22 days ago

Apply formatting

jesus-orozcopushed 1 commit to pathways_trillium • e9d8341…357a322 • 
on Dec 12, 2024

Refactor jobset to align with new pathways structure

jesus-orozcopushed 1 commit to pathways_trillium • 2e0dd4b…e9d8341 • 
on Dec 12, 2024

Update compiler options validation

jesus-orozcopushed 2 commits to feature/v6e • a23a918…9195b9e • 
on Dec 11, 2024

Add remat policy for fuji-70B on tpu-v6e

jesus-orozcocreated feature/v6e • a23a918 • 
on Dec 11, 2024

Support v6e (apple#879)

jesus-orozcopushed 8 commits to main • f61a269…73625c9 • 
on Dec 11, 2024

dump xla flags to gcs

jesus-orozcopushed 1 commit to pathways_trillium • 735ff37…2e0dd4b • 
on Dec 9, 2024

Enable host network on gke pod spec

jesus-orozcopushed 1 commit to feature/pathways_workload • ce5f1a2…a37b7ca • 
on Dec 9, 2024

Merge branch 'apple:main' into pathways_trillium

jesus-orozcopushed 9 commits to pathways_trillium • f387807…735ff37 • 
on Dec 6, 2024

Support orbax state builder (apple#866)

jesus-orozcopushed 10 commits to main • d0edead…f61a269 • 
on Dec 6, 2024

update pathways jobset definition

jesus-orozcopushed 3 commits to pathways_trillium • 0c33403…f387807 • 
on Dec 5, 2024

Temp change to use Orbax checkpointer for Fuji

jiya-zhangcreated maggiejz/pathways-trillium • 86308df • 
on Nov 27, 2024

Launch pathways on trainer_main

jesus-orozcopushed 3 commits to pathways_trillium • 4f58291…0c33403 • 
on Nov 25, 2024

Merge branch 'apple:main' into pathways_trillium

jesus-orozcopushed 3 commits to pathways_trillium • bf82554…4f58291 • 
on Nov 25, 2024

merge trillium changes

jesus-orozcopushed 24 commits to pathways_trillium • 5704ce5…bf82554 • 
on Nov 22, 2024

pathways jobset updates

jesus-orozcocreated pathways_trillium • 5704ce5 • 
on Nov 22, 2024

Deleted branch

jesus-orozcodeleted pathways_trillium_testing • 
on Nov 22, 2024

snapshot (apple#854)

jesus-orozcocreated pathways_trillium_testing • d0edead • 
on Nov 22, 2024

snapshot (apple#854)

jesus-orozcopushed 2 commits to main • 2803b36…d0edead • 
on Nov 22, 2024

Fix RLHF slowdown in attention multi steps extend_step. (apple#849)

jesus-orozcopushed 4 commits to main • 594313d…2803b36 • 
on Nov 19, 2024