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

Rebase tensorflow-alpa & Update jax to 0.3.15 #682

Merged
merged 18 commits into from
Aug 30, 2022
Merged

Rebase tensorflow-alpa & Update jax to 0.3.15 #682

merged 18 commits into from
Aug 30, 2022
+821 −1,023

Conversation

merrymercy
Copy link
Member

@merrymercy merrymercy commented Aug 30, 2022 •
edited
Loading

  • Rebase tensorflow and update jax to 0.3.15.
    • The major change is the lowering flow from Jaxpr to MLIR to HLO
    • identity_marker is deprecated. We use the built-in HLO OptimizationBarrier to implement this.

I will add instructions for syncing local repos later.

@merrymercy merrymercy merged commit 73be80c into main Aug 30, 2022
@merrymercy merrymercy deleted the rebase branch August 30, 2022 15:28
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.

None yet
1 participant
@merrymercy