API Docs
Pompon
Python package for potential optimizer in matrix product operator
| model.Model | Abstract Model class |
| model.NNMPO | Neural Network Matrix Product Operator |
| sop.SumOfProducts | Function given by sum of products |
| sop.OneBody | Function given by sum of one-body functions |
| layers.parameters | |
| layers.parameters.Parameter | Parameter abstract object |
| layers.tensor | |
| layers.tensor.Tensor | Tensor class support “leg_names” for tensor network |
| layers.tensor.Core | TT-Core tensor |
| layers.tensor.TwodotCore | Two-dot tensor |
| layers.tensor.LeftBlockBatch | Left blocks for batch are calculated |
| layers.tensor.RightBlockBatch | Right blocks for batch are calculated |
| layers.tensor.BasisBatch | Basis batch \(D\) @ \(\phi^{[p]}_{i_p}\). |
| layers.tt.TensorTrain | Tensor Train (TT) class |
| layers.coordinator.Stiefel | Stiefel manifold class |
| layers.coordinator.Coordinator | Coordinator layer class |
| layers.basis.Basis | Basis layer class |
| layers.basis.Phi | Phi (1-Basis) layer class |
| layers.linear.Linear | |
| layers | Layers module |
| layers.activations | Activation functions for neural networks. |
| optimizer | Optimizer module |
| optimizer.sweeper.sweep | Tensor-train sweep optimization |
| optimizer.sweeper.Sweeper | Sweep optimizer for tensor-train |
| optimizer.lin_reg.LinearRegression | |
| optimizer.Optimizer | Base class for optimizers |
| optimizer.adam.Adam | Adam optimizer class |
| optimizer.momentum.Momentum | |
| optimizer.sgd.SGD | Steepest Gradient Descent (SGD) optimizer class |
| dataloader.DataLoader | DataLoader class for mini-batch training. |
| utils | |
| losses |