optim

• FishLeg package
  • Submodules
  • FishLeg.fishleg module
    • FishLeg
      • FishLeg.init_model_aux()
      • FishLeg.pretrain_fish()
      • FishLeg.step()
      • FishLeg.update_aux()
      • FishLeg.warmup_aux()
  • FishLeg.fishleg_layers module
    • FishBatchNorm2d
      • FishBatchNorm2d.Qv()
      • FishBatchNorm2d.affine
      • FishBatchNorm2d.diagQ()
      • FishBatchNorm2d.eps
      • FishBatchNorm2d.momentum
      • FishBatchNorm2d.num_features
      • FishBatchNorm2d.track_running_stats
    • FishConv2d
      • FishConv2d.Qv()
      • FishConv2d.bias
      • FishConv2d.diagQ()
      • FishConv2d.dilation
      • FishConv2d.groups
      • FishConv2d.in_channels
      • FishConv2d.kernel_size
      • FishConv2d.out_channels
      • FishConv2d.output_padding
      • FishConv2d.padding
      • FishConv2d.padding_mode
      • FishConv2d.stride
      • FishConv2d.transposed
      • FishConv2d.warmup()
      • FishConv2d.weight
    • FishLayerNorm
      • FishLayerNorm.Qv()
      • FishLayerNorm.diagQ()
      • FishLayerNorm.elementwise_affine
      • FishLayerNorm.eps
      • FishLayerNorm.normalized_shape
    • FishLinear
      • FishLinear.Qg()
      • FishLinear.Qv()
      • FishLinear.diagQ()
      • FishLinear.in_features
      • FishLinear.out_features
      • FishLinear.save_layer_grad_output()
      • FishLinear.save_layer_input()
      • FishLinear.warmup()
      • FishLinear.weight
  • FishLeg.fishleg_likelihood module
    • BernoulliLikelihood
      • BernoulliLikelihood.draw()
      • BernoulliLikelihood.nll()
    • FishLikelihood
      • FishLikelihood.draw()
      • FishLikelihood.get_parameters()
      • FishLikelihood.nll()
    • FixedGaussianLikelihood
      • FixedGaussianLikelihood.draw()
      • FixedGaussianLikelihood.get_variance
      • FixedGaussianLikelihood.nll()
    • GaussianLikelihood
      • GaussianLikelihood.Qv()
      • GaussianLikelihood.draw()
      • GaussianLikelihood.get_aux_parameters()
      • GaussianLikelihood.get_parameters()
      • GaussianLikelihood.init_aux()
      • GaussianLikelihood.nll()
    • SoftMaxLikelihood
      • SoftMaxLikelihood.draw()
      • SoftMaxLikelihood.nll()
  • FishLeg.utils module
    • get_named_layers_by_regex()
    • recursive_getattr()
    • recursive_setattr()
    • update_dict()
  • Module contents