Neural Tangents
stable

Topics:

  • Stax – infinite networks (NNGP, NTK)
  • Empirical – finite NNGP and NTK
  • Predict – inference with NNGP and NTK or linearized networks
  • Batching – using multiple devices
  • Monte Carlo Sampling
Neural Tangents
  • »
  • Overview: module code

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

© Copyright 2021, Google LLC.. Revision 8863a8b1.

Built with Sphinx using a theme provided by Read the Docs.