taktiny.nnΒΆ
The taktiny.nn module provides base classes, layers, containers, and building blocks for deep neural networks in JAX.
Back to top