Index

A

adeptml.configs module adeptml.ensemble module

adeptml.models module adeptml.train_utils module

B

backward() (adeptml.models.Physics static method) (adeptml.models.Physics_SplitVJP static method) (adeptml.models.Physics_VJP static method)

C

config (adeptml.models.MLP attribute)

F

forward() (adeptml.ensemble.HybridModel method) (adeptml.models.MLP method) (adeptml.models.Physics static method) (adeptml.models.Physics_SplitVJP static method) (adeptml.models.Physics_VJP static method)

forward_func (adeptml.configs.PhysicsConfig attribute)

H

hidden_activation (adeptml.configs.MLPConfig attribute)

HybridConfig (class in adeptml.configs) HybridModel (class in adeptml.ensemble)

J

jacobian_func (adeptml.configs.PhysicsConfig attribute)

M

MLP (class in adeptml.models) MLPConfig (class in adeptml.configs) model_inputs (adeptml.configs.HybridConfig attribute) models (adeptml.configs.HybridConfig attribute)

module adeptml.configs adeptml.ensemble adeptml.models adeptml.train_utils

N

num_hidden_dim (adeptml.configs.MLPConfig attribute) num_hidden_layers (adeptml.configs.MLPConfig attribute)

num_input_dim (adeptml.configs.MLPConfig attribute) num_output_dim (adeptml.configs.MLPConfig attribute)

O

output_activation (adeptml.configs.MLPConfig attribute)

P

Physics (class in adeptml.models) Physics_SplitVJP (class in adeptml.models)

Physics_VJP (class in adeptml.models) PhysicsConfig (class in adeptml.configs)

T

train() (in module adeptml.train_utils)

train_step() (in module adeptml.train_utils)

U

use_split_vjp (adeptml.configs.PhysicsConfig attribute)

use_vjp (adeptml.configs.PhysicsConfig attribute)