GenerativeRL
0.0.1
Tutorials
Installation
Quick Start
API Documentation
grl.agents
grl.algorithms
grl.datasets
grl.generative_models
grl.neural_network
grl.numerical_methods
grl.rl_modules
grl.utils
GenerativeRL
Index
Index
_
|
A
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
L
|
M
|
N
|
O
|
Q
|
S
|
T
|
U
|
V
_
__init__() (grl.agents.GPAgent method)
(grl.agents.QGPOAgent method)
(grl.agents.SRPOAgent method)
(grl.datasets.GPD4RLDataset method)
(grl.datasets.GPDataset method)
(grl.datasets.QGPOD4RLDataset method)
(grl.datasets.QGPODataset method)
(grl.generative_models.DiffusionModel method)
(grl.generative_models.EnergyConditionalDiffusionModel method)
(grl.generative_models.IndependentConditionalFlowModel method)
(grl.generative_models.OptimalTransportConditionalFlowModel method)
(grl.neural_network.ConcatenateLayer method)
(grl.neural_network.ConcatenateMLP method)
(grl.neural_network.DiT method)
(grl.neural_network.DiT1D method)
(grl.neural_network.DiT3D method)
(grl.neural_network.MultiLayerPerceptron method)
(grl.neural_network.TemporalSpatialResidualNet method)
(grl.numerical_methods.DPMSolver method)
(grl.numerical_methods.GaussianConditionalProbabilityPath method)
(grl.numerical_methods.ODE method)
(grl.numerical_methods.ODESolver method)
(grl.numerical_methods.SDE method)
(grl.numerical_methods.SDESolver method)
(grl.rl_modules.DoubleQNetwork method)
(grl.rl_modules.DoubleVNetwork method)
(grl.rl_modules.GymEnvSimulator method)
(grl.rl_modules.OneShotValueFunction method)
(grl.rl_modules.QNetwork method)
(grl.rl_modules.VNetwork method)
A
act() (grl.agents.GPAgent method)
(grl.agents.QGPOAgent method)
(grl.agents.SRPOAgent method)
C
collect_episodes() (grl.rl_modules.GymEnvSimulator method)
collect_steps() (grl.rl_modules.GymEnvSimulator method)
compute_double_q() (grl.rl_modules.DoubleQNetwork method)
compute_double_v() (grl.rl_modules.DoubleVNetwork method)
(grl.rl_modules.OneShotValueFunction method)
compute_mininum_q() (grl.rl_modules.DoubleQNetwork method)
compute_mininum_v() (grl.rl_modules.DoubleVNetwork method)
ConcatenateLayer (class in grl.neural_network)
ConcatenateMLP (class in grl.neural_network)
covariance() (grl.numerical_methods.GaussianConditionalProbabilityPath method)
D
d_covariance_dt() (grl.numerical_methods.GaussianConditionalProbabilityPath method)
d_log_scale_dt() (grl.numerical_methods.GaussianConditionalProbabilityPath method)
d_scale_dt() (grl.numerical_methods.GaussianConditionalProbabilityPath method)
d_std_dt() (grl.numerical_methods.GaussianConditionalProbabilityPath method)
data_prediction_function() (grl.generative_models.DiffusionModel method)
(grl.generative_models.EnergyConditionalDiffusionModel method)
data_prediction_function_with_energy_guidance() (grl.generative_models.EnergyConditionalDiffusionModel method)
diffusion() (grl.numerical_methods.GaussianConditionalProbabilityPath method)
diffusion_squared() (grl.numerical_methods.GaussianConditionalProbabilityPath method)
DiffusionModel (class in grl.generative_models)
DiT (class in grl.neural_network)
DiT1D (class in grl.neural_network)
DiT2D (in module grl.neural_network)
DiT3D (class in grl.neural_network)
DoubleQNetwork (class in grl.rl_modules)
DoubleVNetwork (class in grl.rl_modules)
DPMSolver (class in grl.numerical_methods)
dpo_loss() (grl.generative_models.DiffusionModel method)
drift() (grl.numerical_methods.GaussianConditionalProbabilityPath method)
drift_coefficient() (grl.numerical_methods.GaussianConditionalProbabilityPath method)
E
energy_guidance_loss() (grl.generative_models.EnergyConditionalDiffusionModel method)
EnergyConditionalDiffusionModel (class in grl.generative_models)
evaluate() (grl.rl_modules.GymEnvSimulator method)
F
flow_matching_loss() (grl.generative_models.DiffusionModel method)
(grl.generative_models.EnergyConditionalDiffusionModel method)
(grl.generative_models.IndependentConditionalFlowModel method)
(grl.generative_models.OptimalTransportConditionalFlowModel method)
flow_matching_loss_small_batch_OT_plan() (grl.generative_models.OptimalTransportConditionalFlowModel method)
forward() (grl.neural_network.ConcatenateLayer method)
(grl.neural_network.ConcatenateMLP method)
(grl.neural_network.DiT method)
(grl.neural_network.DiT1D method)
(grl.neural_network.DiT3D method)
(grl.neural_network.MultiLayerPerceptron method)
(grl.neural_network.TemporalSpatialResidualNet method)
(grl.rl_modules.DoubleQNetwork method)
(grl.rl_modules.DoubleVNetwork method)
(grl.rl_modules.OneShotValueFunction method)
(grl.rl_modules.QNetwork method)
(grl.rl_modules.VNetwork method)
forward_sample() (grl.generative_models.DiffusionModel method)
forward_sample_process() (grl.generative_models.DiffusionModel method)
forward_with_cfg() (grl.neural_network.DiT method)
G
GaussianConditionalProbabilityPath (class in grl.numerical_methods)
GPAgent (class in grl.agents)
GPD4RLDataset (class in grl.datasets)
GPDataset (class in grl.datasets)
grl.agents
module
grl.datasets
module
grl.generative_models
module
grl.neural_network
module
grl.numerical_methods
module
grl.rl_modules
module
grl.utils
module
GymEnvSimulator (class in grl.rl_modules)
H
HalfLogSNR() (grl.numerical_methods.GaussianConditionalProbabilityPath method)
I
IndependentConditionalFlowModel (class in grl.generative_models)
initialize_weights() (grl.neural_network.DiT method)
(grl.neural_network.DiT1D method)
(grl.neural_network.DiT3D method)
integrate() (grl.numerical_methods.DPMSolver method)
(grl.numerical_methods.ODESolver method)
(grl.numerical_methods.SDESolver method)
InverseHalfLogSNR() (grl.numerical_methods.GaussianConditionalProbabilityPath method)
L
log_scale() (grl.numerical_methods.GaussianConditionalProbabilityPath method)
M
module
grl.agents
grl.datasets
grl.generative_models
grl.neural_network
grl.numerical_methods
grl.rl_modules
grl.utils
MultiLayerPerceptron (class in grl.neural_network)
N
noise_function() (grl.generative_models.DiffusionModel method)
(grl.generative_models.EnergyConditionalDiffusionModel method)
noise_function_with_energy_guidance() (grl.generative_models.EnergyConditionalDiffusionModel method)
O
ODE (class in grl.numerical_methods)
ODESolver (class in grl.numerical_methods)
OneShotValueFunction (class in grl.rl_modules)
OptimalTransportConditionalFlowModel (class in grl.generative_models)
Q
QGPOAgent (class in grl.agents)
QGPOD4RLDataset (class in grl.datasets)
QGPODataset (class in grl.datasets)
QNetwork (class in grl.rl_modules)
S
sample() (grl.generative_models.DiffusionModel method)
(grl.generative_models.EnergyConditionalDiffusionModel method)
(grl.generative_models.IndependentConditionalFlowModel method)
(grl.generative_models.OptimalTransportConditionalFlowModel method)
sample_forward_process() (grl.generative_models.DiffusionModel method)
(grl.generative_models.EnergyConditionalDiffusionModel method)
(grl.generative_models.IndependentConditionalFlowModel method)
(grl.generative_models.OptimalTransportConditionalFlowModel method)
sample_forward_process_with_fixed_x() (grl.generative_models.DiffusionModel method)
(grl.generative_models.EnergyConditionalDiffusionModel method)
sample_with_fixed_x() (grl.generative_models.DiffusionModel method)
(grl.generative_models.EnergyConditionalDiffusionModel method)
sample_with_fixed_x_without_energy_guidance() (grl.generative_models.EnergyConditionalDiffusionModel method)
sample_without_energy_guidance() (grl.generative_models.EnergyConditionalDiffusionModel method)
scale() (grl.numerical_methods.GaussianConditionalProbabilityPath method)
score_function() (grl.generative_models.DiffusionModel method)
(grl.generative_models.EnergyConditionalDiffusionModel method)
score_function_with_energy_guidance() (grl.generative_models.EnergyConditionalDiffusionModel method)
score_matching_loss() (grl.generative_models.DiffusionModel method)
(grl.generative_models.EnergyConditionalDiffusionModel method)
SDE (class in grl.numerical_methods)
SDESolver (class in grl.numerical_methods)
set_seed() (in module grl.utils)
SRPOAgent (class in grl.agents)
std() (grl.numerical_methods.GaussianConditionalProbabilityPath method)
T
TemporalSpatialResidualNet (class in grl.neural_network)
U
unpatchify() (grl.neural_network.DiT method)
(grl.neural_network.DiT3D method)
V
v_loss() (grl.rl_modules.OneShotValueFunction method)
velocity_function() (grl.generative_models.DiffusionModel method)
(grl.generative_models.EnergyConditionalDiffusionModel method)
VNetwork (class in grl.rl_modules)