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
  • »
  • Python Module Index

Python Module Index

n
 
n
- neural_tangents
    neural_tangents._src.batching
    neural_tangents._src.empirical
    neural_tangents._src.monte_carlo
    neural_tangents._src.predict
    neural_tangents.stax

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

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