Skip to content
This repository was archived by the owner on Oct 19, 2024. It is now read-only.

Rebase ALPA to JAX 0.4.10 - #926

Draft
yhtang wants to merge 7 commits into
alpa-projects:devfrom
yhtang:rebase-jax-0.4.10
Draft

Rebase ALPA to JAX 0.4.10#926
yhtang wants to merge 7 commits into
alpa-projects:devfrom
yhtang:rebase-jax-0.4.10

Conversation

@yhtang

@yhtang yhtang commented May 12, 2023

Copy link
Copy Markdown

This is still a work-in-progress.

  • Migrate from tensorflow to OpenXLA
    • migrate all delta between tensorflow-alpa and tensorflow upstream to xla-alpa
    • build ALPA's jaxlib wheel
  • Rebase ALPA's Python code
    • Fix frontend-backend compatibility

@yhtang

yhtang commented May 12, 2023

Copy link
Copy Markdown
Author

Adding @nouiz @ZYHowell @gjoliver @merrymercy @mjsML for viz

@ZYHowell
ZYHowell changed the base branch from main to dev May 15, 2023 02:14
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant