neuromancer.slim package

Subpackages

• neuromancer.slim.butterfly package
  • Submodules
    • neuromancer.slim.butterfly.benchmark module
    • neuromancer.slim.butterfly.butterfly module
      • Butterfly
    • neuromancer.slim.butterfly.butterfly_multiply module
      • ButterflyFactorMult
      • ButterflyMult
      • ButterflyMultInplace
      • ButterflyMultUntied
      • butterfly_mult()
      • butterfly_mult_factors()
      • butterfly_mult_torch()
      • butterfly_mult_untied()
      • butterfly_mult_untied_torch()
    • neuromancer.slim.butterfly.complex_utils module
      • ComplexMatmulNp
      • ComplexMul
      • Conjugate
      • complex_matmul_torch()
      • complex_mul_numpy()
      • complex_mul_torch()
      • conjugate_torch()
      • cupy2torch()
      • real_to_complex()
      • test_complex_mm()
      • test_complex_mul()
      • torch2cupy()
      • torch2numpy()
    • neuromancer.slim.butterfly.permutation module
      • FixedPermutation
      • Permutation
      • PermutationFactor
    • neuromancer.slim.butterfly.permutation_multiply module
      • PermutationFactorEvenOddMult
      • PermutationFactorReverseMult
      • permutation_mult()
      • permutation_mult_factors()
      • permutation_mult_single()
      • permutation_mult_single_factor()
      • permutation_mult_single_factor_torch()
      • permutation_mult_torch()
    • neuromancer.slim.butterfly.utils module
      • bitreversal_permutation()
  • Module contents

Submodules

• neuromancer.slim.bench module
• neuromancer.slim.linear module
  • BoundedNormLinear
    • BoundedNormLinear.reg_error()
  • ButterflyLinear
    • ButterflyLinear.effective_W()
    • ButterflyLinear.forward()
  • DampedSkewSymmetricLinear
    • DampedSkewSymmetricLinear.effective_W()
  • GershgorinLinear
    • GershgorinLinear.effective_W()
  • Hprod()
  • IdentityGradReLU
    • IdentityGradReLU.backward()
    • IdentityGradReLU.forward()
  • IdentityInitLinear
  • IdentityLinear
  • L0Linear
    • L0Linear.cdf_qz()
    • L0Linear.effective_W()
    • L0Linear.get_eps()
    • L0Linear.quantile_concrete()
    • L0Linear.reg_error()
  • LassoLinear
    • LassoLinear.effective_W()
    • LassoLinear.forward()
    • LassoLinear.reg_error()
  • LassoLinearRELU
    • LassoLinearRELU.effective_W()
    • LassoLinearRELU.reg_error()
  • LeftStochasticLinear
    • LeftStochasticLinear.effective_W()
  • Linear
    • Linear.effective_W()
    • Linear.forward()
  • LinearBase
    • LinearBase.device
    • LinearBase.effective_W()
    • LinearBase.eig()
    • LinearBase.forward()
    • LinearBase.reg_error()
  • NonNegativeLinear
    • NonNegativeLinear.effective_W()
  • OrthogonalLinear
    • OrthogonalLinear.effective_W()
    • OrthogonalLinear.forward()
  • PSDLinear
    • PSDLinear.effective_W()
  • PerronFrobeniusLinear
    • PerronFrobeniusLinear.effective_W()
  • PowerBoundLinear
    • PowerBoundLinear.effective_W()
    • PowerBoundLinear.eig_v_estimate()
    • PowerBoundLinear.reg_error()
  • RightStochasticLinear
    • RightStochasticLinear.effective_W()
  • SVDLinear
    • SVDLinear.effective_W()
    • SVDLinear.orthogonal_error()
    • SVDLinear.reg_error()
  • SVDLinearLearnBounds
  • SchurDecompositionLinear
    • SchurDecompositionLinear.build_T()
    • SchurDecompositionLinear.effective_W()
    • SchurDecompositionLinear.reg_error()
  • SkewSymmetricLinear
    • SkewSymmetricLinear.effective_W()
  • SpectralLinear
    • SpectralLinear.Sigma()
    • SpectralLinear.Umultiply()
    • SpectralLinear.Vmultiply()
    • SpectralLinear.effective_W()
    • SpectralLinear.forward()
  • SplitLinear
    • SplitLinear.effective_W()
  • SquareLinear
    • SquareLinear.effective_W()
  • StableSplitLinear
    • StableSplitLinear.effective_W()
  • SymmetricLinear
    • SymmetricLinear.effective_W()
  • SymmetricSVDLinear
  • SymmetricSpectralLinear
  • SymplecticLinear
    • SymplecticLinear.effective_W()
  • TrivialNullSpaceLinear
    • TrivialNullSpaceLinear.effective_W()
• neuromancer.slim.rnn module
  • RNN
    • RNN.forward()
    • RNN.reg_error()
  • RNNCell
    • RNNCell.forward()
    • RNNCell.reg_error()

Module contents