Skip to content

TorchEBM

Cuda

Initializing search

soran-ghaderi/torchebm

Home
Getting Started
API Reference
Examples
Developer Guide
Blog
FAQ

TorchEBM

soran-ghaderi/torchebm

Home
Getting Started
Getting Started
- Introduction
- Installation
- Guides
  Guides
API Reference
API Reference
- Core
  Core
  - Base_energy_function
    
    Base_energy_function
    
    AckleyEnergy
    
    BaseEnergyFunction
    
    DoubleWellEnergy
    
    GaussianEnergy
    
    HarmonicEnergy
    
    RastriginEnergy
    
    RosenbrockEnergy
  - Base_evaluator
  - Base_loss
    
    Base_loss
    
    BaseContrastiveDivergence
    
    BaseLoss
  - Base_metric
    
    Base_metric
    
    BaseMetric
    
    EnergySampleMetric
    
    SampleQualityMetric
  - Base_optimizer
    
    Base_optimizer
    
    Optimizer
  - Base_sampler
    
    Base_sampler
    
    BaseSampler
  - Base_trainer
    
    Base_trainer
    
    ContrastiveDivergenceTrainer
- Cuda Cuda
  Table of contents
  - Contents
  - API Reference
    
    cuda
- Datasets
  Datasets
  - Generators
    
    Generators
    
    BaseSyntheticDataset
    
    CheckerboardDataset
    
    CircleDataset
    
    EightGaussiansDataset
    
    GaussianMixtureDataset
    
    GridDataset
    
    PinwheelDataset
    
    SwissRollDataset
    
    TwoMoonsDataset
- Losses
  Losses
  - Contrastive_divergence
    
    Contrastive_divergence
    
    ContrastiveDivergence
    
    ParallelTemperingCD
    
    PersistentContrastiveDivergence
  - Score_matching
    
    Score_matching
    
    ScoreMatching
- Models
  Models
  - Base_model
    
    Base_model
    
    BaseModel
- Samplers
  Samplers
  - Hmc
    
    Hmc
    
    HamiltonianMonteCarlo
  - Langevin_dynamics
    
    Langevin_dynamics
    
    LangevinDynamics
- Utils
  Utils
  - Visualization
Examples
Examples
- Energy Functions
  Energy Functions
- Datasets
  Datasets
- Samplers
  Samplers
- Training EBMs
  Training EBMs
  - Learning Gaussian Mixture
- Visualization
  Visualization
Developer Guide
Developer Guide
- Contributing
- Commit Conventions
- Development Setup
- Code Style
- Testing Guide
- Project Structure
- Design Principles
- Core Components
- Implementation Details
  Implementation Details
  - Energy Functions
  - Samplers
- Architecture
- API Design
- API Generation
- Performance Optimization
Blog
FAQ

Table of contents

Contents
API Reference
- cuda

Torchebm > Cuda¶

Contents¶

API Reference¶

torchebm.cuda ¶

CUDA-accelerated implementations of key operations for improved performance.

April 18, 2025 April 6, 2025

ContrastiveDivergenceTrainer

Torchebm > Datasets

Copyright © 2025 Soran Ghaderi. Licensed under the MIT License.