flax
Avoid depending on JAX internals, which are about to change (see cl/681582933).
#4321

Open

Avoid depending on JAX internals, which are about to change (see cl/681582933). #4321

copybara-service wants to merge 1 commit into main from test_688953662

copybara-service

dougalm

Avoid depending on JAX internals, which are about to change (see cl/6…

4d057dd1

Login to write a write a comment.

Login via GitHub

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone