Project Awesome project awesome

Neural Network Libraries > Neural Tangents

High-level API for specifying neural networks of both finite and infinite width.

Package 2.4k stars Archived GitHub
Back to JAX