flax
18be4d4d - Add support for JAX dynamic stack-based named_call.

Commit

3 years ago

Add support for JAX dynamic stack-based named_call. PiperOrigin-RevId: 443808300

References

#2063 - Add support for JAX dynamic stack-based named_call.

Author

levskaya

levskaya

Committer

a-googler

a-googler

Parents

Loading