Module pytorch_api

Module pytorch_api 

Source
Expand description

PyTorch-like API for quantum machine learning models

This module provides a familiar PyTorch-style interface for building, training, and deploying quantum ML models, making it easier for classical ML practitioners to adopt quantum algorithms.

Structs§

CosineAnnealingLR
Cosine annealing learning rate scheduler
ExponentialLR
Exponential learning rate scheduler
LSTMState
LSTM cell state
MemoryDataLoader
Simple in-memory data loader
Parameter
Quantum parameter wrapper
PositionalEncoding
Positional encoding for transformers
QuantumActivation
Quantum activation functions
QuantumAdaptiveAvgPool2d
Adaptive average pooling 2D
QuantumAvgPool2d
Average pooling 2D
QuantumBCELoss
Binary Cross Entropy Loss
QuantumBCEWithLogitsLoss
Binary Cross Entropy with Logits Loss
QuantumBatchNorm1d
Batch normalization layer
QuantumConv1d
1D Convolution layer
QuantumConv2d
Quantum convolutional layer
QuantumConv3d
3D Convolution layer
QuantumCrossEntropyLoss
Cross Entropy loss
QuantumDropout
Dropout layer
QuantumDropout2d
Dropout2d for convolutional layers
QuantumEmbedding
Embedding layer for discrete inputs
QuantumExtendedActivation
Extended activation layer
QuantumGRU
GRU layer
QuantumKLDivLoss
Kullback-Leibler Divergence Loss
QuantumL1Loss
L1 Loss (Mean Absolute Error)
QuantumLSTM
LSTM layer
QuantumLayerNorm
Layer normalization
QuantumLinear
Quantum linear layer
QuantumMSELoss
Mean Squared Error loss
QuantumMaxPool2d
Max pooling 2D
QuantumMultiheadAttention
Multi-head attention layer
QuantumNLLLoss
Negative Log Likelihood Loss
QuantumSequential
Sequential container for quantum modules
QuantumSmoothL1Loss
Smooth L1 Loss (Huber Loss)
QuantumTrainer
Training utilities
QuantumTransformerEncoderLayer
Transformer encoder layer
ReduceLROnPlateau
ReduceLROnPlateau scheduler
StepLR
Step learning rate scheduler
TrainingHistory
Training history

Enums§

ActivationType
Activation function types
ExtendedActivation
Extended activation function types
InitType
Parameter initialization types

Traits§

DataLoader
Data loader trait
LRScheduler
Learning rate scheduler trait
QuantumLoss
Loss functions for quantum ML
QuantumModule
Base trait for all quantum ML modules

Functions§

init_weights
Initialize parameters with specified method