flax
0c5232e6 - Add support for PyTrees with None leaves. Add additional unit tests.

Commit
3 years ago
Add support for PyTrees with None leaves. Add additional unit tests. PiperOrigin-RevId: 494218037
Author
Committer
Parents
Loading