Se3 Molecular

Module: generative_models.models.flow.se3_molecular

Source: generative_models/models/flow/se3_molecular.py

Overview

SE(3)-Equivariant Molecular Flow for conformation generation.

Classes

SE3CouplingLayer

class SE3CouplingLayer

SE3EquivariantLayer

class SE3EquivariantLayer

SE3MolecularFlow

class SE3MolecularFlow

Functions

call

def __call__()

call

def __call__()

init

def __init__()

init

def __init__()

init

def __init__()

generate

def generate()

log_prob

def log_prob()

loss_fn

def loss_fn()

sample

def sample()

Module Statistics

• Classes: 3
• Functions: 9
• Imports: 5