Add type interface for `jax.nn` - SemanticDiff

Commit

309 days ago

Add type interface for `jax.nn` Re-landing of #29410 with fixes. PiperOrigin-RevId: 784345067

References

#30196 - Add type interface for `jax.nn`

Author

Jake VanderPlas

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading