Neural Networks

Functions and modules to learn linear operators via neural networks.

• nn
  • Table of Contents
  • Regressors (see also kernel regressors)
    • ridge_least_squares()
    • eig()
    • evaluate_eigenfunction()
  • Loss Functions
    • L2ContrastiveLoss
      • L2ContrastiveLoss.forward()
    • KLContrastiveLoss
      • KLContrastiveLoss.forward()
    • VampLoss
      • VampLoss.forward()
    • DPLoss
      • DPLoss.forward()
  • Modules
    • MLP
    • ResNet
    • SimNorm
    • EMACovariance
• nn.functional
  • Table of Contents
  • Loss Functions
    • l2_contrastive_loss()
    • kl_contrastive_loss()
    • vamp_loss()
    • dp_loss()
  • Regularization Functions
    • orthonormal_fro_reg()
    • orthonormal_logfro_reg()
• nn.stats
  • covariance()
  • cov_norm_squared_unbiased()
  • cross_cov_norm_squared_unbiased()
  • whitening()
• nn.linalg
  • sqrtmh()