All modules for which code is available
- neural_tangents._src.batching
- neural_tangents._src.empirical
- neural_tangents._src.monte_carlo
- neural_tangents._src.predict
- neural_tangents._src.stax.branching
- neural_tangents._src.stax.combinators
- neural_tangents._src.stax.elementwise
- neural_tangents._src.stax.linear
- neural_tangents._src.stax.requirements
- neural_tangents._src.utils.kernel
- neural_tangents._src.utils.typing
- neural_tangents.experimental.empirical_tf.empirical
- typing