jax
09dfd474 - Merge pull request #4631 from hawkinsp:src

Commit

5 years ago

Merge pull request #4631 from hawkinsp:src PiperOrigin-RevId: 337673701

References

#4631 - Move jax.nn implementation into jax._src.nn.

Author

a-googler

a-googler

Parents

Loading