Index

A

accel (jaxdem.State attribute) (jaxdem.state.State attribute) action (jaxdem.rl.TrajectoryData attribute) action_space_shape (jaxdem.rl.Environment attribute) (jaxdem.rl.environments.Environment attribute) action_space_size (jaxdem.rl.Environment attribute) (jaxdem.rl.environments.Environment attribute) ActionSpace (class in jaxdem.rl) (class in jaxdem.rl.actionSpaces) actor (jaxdem.rl.models.ActorCritic attribute) (jaxdem.rl.models.LSTM.LSTMActorCritic attribute) (jaxdem.rl.models.LSTMActorCritic attribute) (jaxdem.rl.models.MLP.ActorCritic attribute) (jaxdem.rl.models.MLP.SharedActorCritic attribute) (jaxdem.rl.models.SharedActorCritic attribute) actor_torso (jaxdem.rl.models.ActorCritic attribute) (jaxdem.rl.models.MLP.ActorCritic attribute)

ActorCritic (class in jaxdem.rl.models) (class in jaxdem.rl.models.MLP) add() (jaxdem.State static method) (jaxdem.state.State static method) advantage (jaxdem.rl.TrajectoryData attribute) advantage_c_clip (jaxdem.rl.Trainer attribute) advantage_gamma (jaxdem.rl.Trainer attribute) advantage_lambda (jaxdem.rl.Trainer attribute) advantage_rho_clip (jaxdem.rl.Trainer attribute) anchor (jaxdem.Domain attribute) (jaxdem.domains.Domain attribute) angle() (in module jaxdem.utils) (in module jaxdem.utils.angles) angle_x() (in module jaxdem.utils) (in module jaxdem.utils.angles) anneal_importance_sampling_beta (jaxdem.rl.trainers.PPOTrainer attribute) (jaxdem.rl.trainers.PPOtrainer.PPOTrainer attribute)

B

batch_size (jaxdem.State property) (jaxdem.state.State property) bij (jaxdem.rl.models.ActorCritic attribute) (jaxdem.rl.models.LSTM.LSTMActorCritic attribute) (jaxdem.rl.models.LSTMActorCritic attribute) (jaxdem.rl.models.MLP.ActorCritic attribute) (jaxdem.rl.models.MLP.SharedActorCritic attribute) (jaxdem.rl.models.SharedActorCritic attribute) binary (jaxdem.VTKWriter attribute) (jaxdem.writers.VTKWriter attribute) (jaxdem.writers.vtkWriter.VTKWriter attribute) block_until_ready() (jaxdem.CheckpointLoader method) (jaxdem.CheckpointModelLoader method) (jaxdem.CheckpointModelWriter method) (jaxdem.CheckpointWriter method) (jaxdem.VTKWriter method) (jaxdem.writers.CheckpointLoader method) (jaxdem.writers.CheckpointModelLoader method) (jaxdem.writers.CheckpointModelWriter method) (jaxdem.writers.checkpoints.CheckpointLoader method) (jaxdem.writers.checkpoints.CheckpointModelLoader method) (jaxdem.writers.checkpoints.CheckpointModelWriter method) (jaxdem.writers.checkpoints.CheckpointWriter method) (jaxdem.writers.CheckpointWriter method) (jaxdem.writers.VTKWriter method) (jaxdem.writers.vtkWriter.VTKWriter method)

box_size (jaxdem.Domain attribute) (jaxdem.domains.Domain attribute) BoxSpace (class in jaxdem.rl.actionSpaces) (class in jaxdem.rl.actionSpaces.boxSpace)

C

cell (jaxdem.rl.models.LSTM.LSTMActorCritic attribute) (jaxdem.rl.models.LSTMActorCritic attribute) checkpointer (jaxdem.CheckpointLoader attribute) (jaxdem.CheckpointModelLoader attribute) (jaxdem.CheckpointModelWriter attribute) (jaxdem.CheckpointWriter attribute) (jaxdem.writers.CheckpointLoader attribute) (jaxdem.writers.CheckpointModelLoader attribute) (jaxdem.writers.CheckpointModelWriter attribute) (jaxdem.writers.checkpoints.CheckpointLoader attribute) (jaxdem.writers.checkpoints.CheckpointModelLoader attribute) (jaxdem.writers.checkpoints.CheckpointModelWriter attribute) (jaxdem.writers.checkpoints.CheckpointWriter attribute) (jaxdem.writers.CheckpointWriter attribute) CheckpointLoader (class in jaxdem) (class in jaxdem.writers) (class in jaxdem.writers.checkpoints) CheckpointModelLoader (class in jaxdem) (class in jaxdem.writers) (class in jaxdem.writers.checkpoints) CheckpointModelWriter (class in jaxdem) (class in jaxdem.writers) (class in jaxdem.writers.checkpoints) CheckpointWriter (class in jaxdem) (class in jaxdem.writers) (class in jaxdem.writers.checkpoints) clean (jaxdem.CheckpointModelWriter attribute) (jaxdem.VTKWriter attribute) (jaxdem.writers.CheckpointModelWriter attribute) (jaxdem.writers.checkpoints.CheckpointModelWriter attribute) (jaxdem.writers.VTKWriter attribute) (jaxdem.writers.vtkWriter.VTKWriter attribute) clip_action_env() (in module jaxdem.rl) (in module jaxdem.rl.envWrappers) close() (jaxdem.CheckpointLoader method) (jaxdem.CheckpointModelLoader method) (jaxdem.CheckpointModelWriter method) (jaxdem.CheckpointWriter method) (jaxdem.VTKWriter method) (jaxdem.writers.CheckpointLoader method) (jaxdem.writers.CheckpointModelLoader method) (jaxdem.writers.CheckpointModelWriter method) (jaxdem.writers.checkpoints.CheckpointLoader method) (jaxdem.writers.checkpoints.CheckpointModelLoader method) (jaxdem.writers.checkpoints.CheckpointModelWriter method) (jaxdem.writers.checkpoints.CheckpointWriter method) (jaxdem.writers.CheckpointWriter method) (jaxdem.writers.VTKWriter method) (jaxdem.writers.vtkWriter.VTKWriter method)

Collider (class in jaxdem) (class in jaxdem.colliders) collider (jaxdem.System attribute) (jaxdem.system.System attribute) compute_advantages() (jaxdem.rl.Trainer static method) compute_force() (jaxdem.Collider static method) (jaxdem.colliders.Collider static method) (jaxdem.colliders.naive.NaiveSimulator static method) (jaxdem.colliders.NaiveSimulator static method) compute_potential_energy() (jaxdem.Collider static method) (jaxdem.colliders.Collider static method) (jaxdem.colliders.naive.NaiveSimulator static method) (jaxdem.colliders.NaiveSimulator static method) Create() (jaxdem.Domain class method) (jaxdem.domains.Domain class method) create() (jaxdem.Factory class method) (jaxdem.factory.Factory class method) Create() (jaxdem.rl.environments.multi_navigator.MultiNavigator class method) (jaxdem.rl.environments.MultiNavigator class method) (jaxdem.rl.environments.single_navigator.SingleNavigator class method) (jaxdem.rl.environments.SingleNavigator class method) (jaxdem.rl.trainers.PPOTrainer class method) (jaxdem.rl.trainers.PPOtrainer.PPOTrainer class method) create() (jaxdem.State static method) (jaxdem.state.State static method) (jaxdem.System static method) (jaxdem.system.System static method) critic (jaxdem.rl.models.ActorCritic attribute) (jaxdem.rl.models.LSTM.LSTMActorCritic attribute) (jaxdem.rl.models.LSTMActorCritic attribute) (jaxdem.rl.models.MLP.ActorCritic attribute) (jaxdem.rl.models.MLP.SharedActorCritic attribute) (jaxdem.rl.models.SharedActorCritic attribute) critic_torso (jaxdem.rl.models.ActorCritic attribute) (jaxdem.rl.models.MLP.ActorCritic attribute)

D

decode_callable() (in module jaxdem.utils) (in module jaxdem.utils.serialization) dim (jaxdem.State property) (jaxdem.state.State property) (jaxdem.System attribute) (jaxdem.system.System attribute) DirectEuler (class in jaxdem.integrators) (class in jaxdem.integrators.direct_euler) directory (jaxdem.CheckpointLoader attribute) (jaxdem.CheckpointModelLoader attribute) (jaxdem.CheckpointModelWriter attribute) (jaxdem.CheckpointWriter attribute) (jaxdem.VTKWriter attribute) (jaxdem.writers.CheckpointLoader attribute) (jaxdem.writers.CheckpointModelLoader attribute) (jaxdem.writers.CheckpointModelWriter attribute) (jaxdem.writers.checkpoints.CheckpointLoader attribute) (jaxdem.writers.checkpoints.CheckpointModelLoader attribute) (jaxdem.writers.checkpoints.CheckpointModelWriter attribute) (jaxdem.writers.checkpoints.CheckpointWriter attribute) (jaxdem.writers.CheckpointWriter attribute) (jaxdem.writers.VTKWriter attribute) (jaxdem.writers.vtkWriter.VTKWriter attribute)

displacement() (jaxdem.Domain static method) (jaxdem.domains.Domain static method) (jaxdem.domains.free.FreeDomain static method) (jaxdem.domains.FreeDomain static method) (jaxdem.domains.periodic.PeriodicDomain static method) (jaxdem.domains.PeriodicDomain static method) (jaxdem.domains.reflect.ReflectDomain static method) (jaxdem.domains.ReflectDomain static method) Domain (class in jaxdem) (class in jaxdem.domains) domain (jaxdem.System attribute) (jaxdem.system.System attribute) done (jaxdem.rl.TrajectoryData attribute) done() (jaxdem.rl.Environment static method) (jaxdem.rl.environments.Environment static method) (jaxdem.rl.environments.multi_navigator.MultiNavigator static method) (jaxdem.rl.environments.MultiNavigator static method) (jaxdem.rl.environments.single_navigator.SingleNavigator static method) (jaxdem.rl.environments.SingleNavigator static method) dt (jaxdem.System attribute) (jaxdem.system.System attribute)

E

Elastic (class in jaxdem.materials) (class in jaxdem.materials.elasticMats) ElasticFriction (class in jaxdem.materials) (class in jaxdem.materials.elasticMats) encode_callable() (in module jaxdem.utils) (in module jaxdem.utils.serialization) encoder (jaxdem.rl.models.LSTM.LSTMActorCritic attribute) (jaxdem.rl.models.LSTMActorCritic attribute) energy() (jaxdem.ForceModel static method) (jaxdem.ForceRouter static method) (jaxdem.forces.ForceModel static method) (jaxdem.forces.ForceRouter static method) (jaxdem.forces.law_combiner.LawCombiner static method) (jaxdem.forces.LawCombiner static method) (jaxdem.forces.router.ForceRouter static method) (jaxdem.forces.spring.SpringForce static method) (jaxdem.forces.SpringForce static method) (jaxdem.LawCombiner static method)

env (jaxdem.rl.Trainer attribute) env_params (jaxdem.rl.Environment attribute) (jaxdem.rl.environments.Environment attribute) Environment (class in jaxdem.rl) (class in jaxdem.rl.environments) epoch() (jaxdem.rl.Trainer static method) (jaxdem.rl.trainers.PPOTrainer static method) (jaxdem.rl.trainers.PPOtrainer.PPOTrainer static method)

F

Factory (class in jaxdem) (class in jaxdem.factory) fixed (jaxdem.State attribute) (jaxdem.state.State attribute) force() (jaxdem.ForceModel static method) (jaxdem.ForceRouter static method) (jaxdem.forces.ForceModel static method) (jaxdem.forces.ForceRouter static method) (jaxdem.forces.law_combiner.LawCombiner static method) (jaxdem.forces.LawCombiner static method) (jaxdem.forces.router.ForceRouter static method) (jaxdem.forces.spring.SpringForce static method) (jaxdem.forces.SpringForce static method) (jaxdem.LawCombiner static method) force_model (jaxdem.System attribute) (jaxdem.system.System attribute) ForceModel (class in jaxdem) (class in jaxdem.forces) ForceRouter (class in jaxdem) (class in jaxdem.forces) (class in jaxdem.forces.router)

forward_and_log_det() (jaxdem.rl.actionSpaces.BoxSpace method) (jaxdem.rl.actionSpaces.boxSpace.BoxSpace method) (jaxdem.rl.actionSpaces.FreeSpace method) (jaxdem.rl.actionSpaces.freeSpace.FreeSpace method) (jaxdem.rl.actionSpaces.MaxNormSpace method) (jaxdem.rl.actionSpaces.maxNormSpace.MaxNormSpace method) forward_log_det_jacobian() (jaxdem.rl.actionSpaces.BoxSpace method) (jaxdem.rl.actionSpaces.boxSpace.BoxSpace method) (jaxdem.rl.actionSpaces.MaxNormSpace method) (jaxdem.rl.actionSpaces.maxNormSpace.MaxNormSpace method) FreeDomain (class in jaxdem.domains) (class in jaxdem.domains.free) FreeSpace (class in jaxdem.rl.actionSpaces) (class in jaxdem.rl.actionSpaces.freeSpace) from_dict() (jaxdem.ForceRouter static method) (jaxdem.forces.ForceRouter static method) (jaxdem.forces.router.ForceRouter static method) from_materials() (jaxdem.materials.MaterialTable static method) (jaxdem.materials.materialTable.MaterialTable static method) (jaxdem.MaterialTable static method)

G

get_effective_property() (jaxdem.material_matchmakers.harmonic.HarmonicMaterialMatchmaker static method) (jaxdem.material_matchmakers.HarmonicMaterialMatchmaker static method) (jaxdem.material_matchmakers.linear.LinearMaterialMatchmaker static method) (jaxdem.material_matchmakers.LinearMaterialMatchmaker static method) (jaxdem.material_matchmakers.MaterialMatchmaker static method) (jaxdem.MaterialMatchmaker static method)

graphdef (jaxdem.rl.Trainer attribute) graphstate (jaxdem.rl.Trainer attribute) grid_state() (in module jaxdem.utils) (in module jaxdem.utils.gridState)

H

HarmonicMaterialMatchmaker (class in jaxdem.material_matchmakers) (class in jaxdem.material_matchmakers.harmonic)

I

ID (jaxdem.State attribute) (jaxdem.state.State attribute) importance_sampling_alpha (jaxdem.rl.trainers.PPOTrainer attribute) (jaxdem.rl.trainers.PPOtrainer.PPOTrainer attribute) importance_sampling_beta (jaxdem.rl.trainers.PPOTrainer attribute) (jaxdem.rl.trainers.PPOtrainer.PPOTrainer attribute) info() (jaxdem.rl.Environment static method) (jaxdem.rl.environments.Environment static method) initialize() (jaxdem.Integrator static method) (jaxdem.integrators.direct_euler.DirectEuler static method) (jaxdem.integrators.DirectEuler static method) (jaxdem.integrators.Integrator static method) Integrator (class in jaxdem) (class in jaxdem.integrators)

integrator (jaxdem.System attribute) (jaxdem.system.System attribute) inverse_and_log_det() (jaxdem.rl.actionSpaces.BoxSpace method) (jaxdem.rl.actionSpaces.boxSpace.BoxSpace method) (jaxdem.rl.actionSpaces.FreeSpace method) (jaxdem.rl.actionSpaces.freeSpace.FreeSpace method) (jaxdem.rl.actionSpaces.MaxNormSpace method) (jaxdem.rl.actionSpaces.maxNormSpace.MaxNormSpace method) is_valid (jaxdem.State property) (jaxdem.state.State property) is_wrapped() (in module jaxdem.rl) (in module jaxdem.rl.envWrappers)

J

jaxdem module jaxdem.colliders module jaxdem.colliders.naive module jaxdem.domains module jaxdem.domains.free module jaxdem.domains.periodic module jaxdem.domains.reflect module jaxdem.factory module jaxdem.forces module jaxdem.forces.law_combiner module jaxdem.forces.router module jaxdem.forces.spring module jaxdem.integrators module jaxdem.integrators.direct_euler module jaxdem.material_matchmakers module jaxdem.material_matchmakers.harmonic module jaxdem.material_matchmakers.linear module jaxdem.materials module jaxdem.materials.elasticMats module jaxdem.materials.materialTable module jaxdem.rl module jaxdem.rl.actionSpaces module jaxdem.rl.actionSpaces.boxSpace module jaxdem.rl.actionSpaces.freeSpace module

jaxdem.rl.actionSpaces.maxNormSpace module jaxdem.rl.environments module jaxdem.rl.environments.multi_navigator module jaxdem.rl.environments.single_navigator module jaxdem.rl.envWrappers module jaxdem.rl.models module jaxdem.rl.models.LSTM module jaxdem.rl.models.MLP module jaxdem.rl.trainers module jaxdem.rl.trainers.PPOtrainer module jaxdem.state module jaxdem.system module jaxdem.utils module jaxdem.utils.angles module jaxdem.utils.gridState module jaxdem.utils.linalg module jaxdem.utils.randomState module jaxdem.utils.serialization module jaxdem.writers module jaxdem.writers.checkpoints module jaxdem.writers.vtkDomainWriter module jaxdem.writers.vtkSpheresWriter module jaxdem.writers.vtkWriter module

K

key (jaxdem.rl.Trainer attribute) kws (jaxdem.rl.ActionSpace property) (jaxdem.rl.actionSpaces.ActionSpace property) (jaxdem.rl.actionSpaces.BoxSpace property) (jaxdem.rl.actionSpaces.boxSpace.BoxSpace property) (jaxdem.rl.actionSpaces.FreeSpace property) (jaxdem.rl.actionSpaces.freeSpace.FreeSpace property) (jaxdem.rl.actionSpaces.MaxNormSpace property) (jaxdem.rl.actionSpaces.maxNormSpace.MaxNormSpace property)

L

latest_step() (jaxdem.CheckpointModelLoader method) (jaxdem.writers.CheckpointModelLoader method) (jaxdem.writers.checkpoints.CheckpointModelLoader method) LawCombiner (class in jaxdem) (class in jaxdem.forces) (class in jaxdem.forces.law_combiner) laws (jaxdem.ForceModel attribute) (jaxdem.forces.ForceModel attribute) LinearMaterialMatchmaker (class in jaxdem.material_matchmakers) (class in jaxdem.material_matchmakers.linear) load() (jaxdem.CheckpointLoader method) (jaxdem.CheckpointModelLoader method) (jaxdem.writers.CheckpointLoader method) (jaxdem.writers.CheckpointModelLoader method) (jaxdem.writers.checkpoints.CheckpointLoader method) (jaxdem.writers.checkpoints.CheckpointModelLoader method) log_prob (jaxdem.rl.TrajectoryData attribute) log_std (jaxdem.rl.Model property) (jaxdem.rl.models.ActorCritic attribute) (jaxdem.rl.models.ActorCritic property) (jaxdem.rl.models.LSTM.LSTMActorCritic attribute) (jaxdem.rl.models.LSTM.LSTMActorCritic property) (jaxdem.rl.models.LSTMActorCritic attribute) (jaxdem.rl.models.LSTMActorCritic property) (jaxdem.rl.models.MLP.ActorCritic attribute) (jaxdem.rl.models.MLP.ActorCritic property) (jaxdem.rl.models.MLP.SharedActorCritic attribute) (jaxdem.rl.models.MLP.SharedActorCritic property) (jaxdem.rl.models.Model property) (jaxdem.rl.models.SharedActorCritic attribute) (jaxdem.rl.models.SharedActorCritic property)

loss_fn() (jaxdem.rl.trainers.PPOTrainer static method) (jaxdem.rl.trainers.PPOtrainer.PPOTrainer static method) lstm_features (jaxdem.rl.models.LSTM.LSTMActorCritic attribute) (jaxdem.rl.models.LSTMActorCritic attribute) LSTMActorCritic (class in jaxdem.rl.models) (class in jaxdem.rl.models.LSTM)

M

mass (jaxdem.State attribute) (jaxdem.state.State attribute) mat_id (jaxdem.State attribute) (jaxdem.state.State attribute) mat_table (jaxdem.System attribute) (jaxdem.system.System attribute) matcher (jaxdem.materials.MaterialTable attribute) (jaxdem.materials.materialTable.MaterialTable attribute) (jaxdem.MaterialTable attribute) Material (class in jaxdem) (class in jaxdem.materials) MaterialMatchmaker (class in jaxdem) (class in jaxdem.material_matchmakers) MaterialTable (class in jaxdem) (class in jaxdem.materials) (class in jaxdem.materials.materialTable) max_num_agents (jaxdem.rl.Environment attribute) (jaxdem.rl.environments.Environment attribute) max_queue_size (jaxdem.VTKWriter attribute) (jaxdem.writers.VTKWriter attribute) (jaxdem.writers.vtkWriter.VTKWriter attribute) max_to_keep (jaxdem.CheckpointModelWriter attribute) (jaxdem.CheckpointWriter attribute) (jaxdem.writers.CheckpointModelWriter attribute) (jaxdem.writers.checkpoints.CheckpointModelWriter attribute) (jaxdem.writers.checkpoints.CheckpointWriter attribute) (jaxdem.writers.CheckpointWriter attribute) max_workers (jaxdem.VTKWriter attribute) (jaxdem.writers.VTKWriter attribute) (jaxdem.writers.vtkWriter.VTKWriter attribute) MaxNormSpace (class in jaxdem.rl.actionSpaces) (class in jaxdem.rl.actionSpaces.maxNormSpace) merge() (jaxdem.State static method) (jaxdem.state.State static method) metadata (jaxdem.rl.Model property) (jaxdem.rl.models.ActorCritic property) (jaxdem.rl.models.LSTM.LSTMActorCritic property) (jaxdem.rl.models.LSTMActorCritic property) (jaxdem.rl.models.MLP.ActorCritic property) (jaxdem.rl.models.MLP.SharedActorCritic property) (jaxdem.rl.models.Model property) (jaxdem.rl.models.SharedActorCritic property) minibatch_size (jaxdem.rl.trainers.PPOTrainer attribute) (jaxdem.rl.trainers.PPOtrainer.PPOTrainer attribute) Model (class in jaxdem.rl) (class in jaxdem.rl.models) model (jaxdem.rl.Trainer property) module jaxdem jaxdem.colliders jaxdem.colliders.naive jaxdem.domains jaxdem.domains.free jaxdem.domains.periodic jaxdem.domains.reflect jaxdem.factory jaxdem.forces jaxdem.forces.law_combiner jaxdem.forces.router jaxdem.forces.spring jaxdem.integrators jaxdem.integrators.direct_euler jaxdem.material_matchmakers jaxdem.material_matchmakers.harmonic jaxdem.material_matchmakers.linear jaxdem.materials jaxdem.materials.elasticMats jaxdem.materials.materialTable jaxdem.rl jaxdem.rl.actionSpaces jaxdem.rl.actionSpaces.boxSpace jaxdem.rl.actionSpaces.freeSpace jaxdem.rl.actionSpaces.maxNormSpace jaxdem.rl.environments jaxdem.rl.environments.multi_navigator jaxdem.rl.environments.single_navigator jaxdem.rl.envWrappers jaxdem.rl.models jaxdem.rl.models.LSTM jaxdem.rl.models.MLP jaxdem.rl.trainers jaxdem.rl.trainers.PPOtrainer jaxdem.state jaxdem.system jaxdem.utils jaxdem.utils.angles jaxdem.utils.gridState jaxdem.utils.linalg jaxdem.utils.randomState jaxdem.utils.serialization jaxdem.writers jaxdem.writers.checkpoints jaxdem.writers.vtkDomainWriter jaxdem.writers.vtkSpheresWriter jaxdem.writers.vtkWriter

mu (jaxdem.materials.ElasticFriction attribute) (jaxdem.materials.elasticMats.ElasticFriction attribute) MultiNavigator (class in jaxdem.rl.environments) (class in jaxdem.rl.environments.multi_navigator)

N

N (jaxdem.State property) (jaxdem.state.State property) NaiveSimulator (class in jaxdem.colliders) (class in jaxdem.colliders.naive) network (jaxdem.rl.models.MLP.SharedActorCritic attribute) (jaxdem.rl.models.SharedActorCritic attribute) new_log_prob (jaxdem.rl.TrajectoryData attribute) num_envs (jaxdem.rl.trainers.PPOTrainer attribute) (jaxdem.rl.trainers.PPOtrainer.PPOTrainer attribute)

num_epochs (jaxdem.rl.trainers.PPOTrainer attribute) (jaxdem.rl.trainers.PPOtrainer.PPOTrainer attribute) num_minibatches (jaxdem.rl.trainers.PPOTrainer attribute) (jaxdem.rl.trainers.PPOtrainer.PPOTrainer attribute) num_segments (jaxdem.rl.trainers.PPOTrainer attribute) (jaxdem.rl.trainers.PPOtrainer.PPOTrainer attribute) num_steps_epoch (jaxdem.rl.trainers.PPOTrainer attribute) (jaxdem.rl.trainers.PPOtrainer.PPOTrainer attribute)

O

obs (jaxdem.rl.TrajectoryData attribute) obs_dim (jaxdem.rl.models.LSTM.LSTMActorCritic attribute) (jaxdem.rl.models.LSTMActorCritic attribute) observation() (jaxdem.rl.Environment static method) (jaxdem.rl.environments.Environment static method) (jaxdem.rl.environments.multi_navigator.MultiNavigator static method) (jaxdem.rl.environments.MultiNavigator static method) (jaxdem.rl.environments.single_navigator.SingleNavigator static method) (jaxdem.rl.environments.SingleNavigator static method)

observation_space_size (jaxdem.rl.Environment attribute) (jaxdem.rl.environments.Environment attribute) optimizer (jaxdem.rl.Trainer property)

P

pair (jaxdem.materials.MaterialTable attribute) (jaxdem.materials.materialTable.MaterialTable attribute) (jaxdem.MaterialTable attribute) periodic (jaxdem.Domain attribute) (jaxdem.domains.Domain attribute) (jaxdem.domains.periodic.PeriodicDomain attribute) (jaxdem.domains.PeriodicDomain attribute) PeriodicDomain (class in jaxdem.domains) (class in jaxdem.domains.periodic) poisson (jaxdem.materials.Elastic attribute) (jaxdem.materials.ElasticFriction attribute) (jaxdem.materials.elasticMats.Elastic attribute) (jaxdem.materials.elasticMats.ElasticFriction attribute)

pos (jaxdem.State attribute) (jaxdem.state.State attribute) ppo_clip_eps (jaxdem.rl.trainers.PPOTrainer attribute) (jaxdem.rl.trainers.PPOtrainer.PPOTrainer attribute) ppo_entropy_coeff (jaxdem.rl.trainers.PPOTrainer attribute) (jaxdem.rl.trainers.PPOtrainer.PPOTrainer attribute) ppo_value_coeff (jaxdem.rl.trainers.PPOTrainer attribute) (jaxdem.rl.trainers.PPOtrainer.PPOTrainer attribute) PPOTrainer (class in jaxdem.rl.trainers) (class in jaxdem.rl.trainers.PPOtrainer) props (jaxdem.materials.MaterialTable attribute) (jaxdem.materials.materialTable.MaterialTable attribute) (jaxdem.MaterialTable attribute)

R

rad (jaxdem.State attribute) (jaxdem.state.State attribute) random_state() (in module jaxdem.utils) (in module jaxdem.utils.randomState) ReflectDomain (class in jaxdem.domains) (class in jaxdem.domains.reflect) register() (jaxdem.Factory class method) (jaxdem.factory.Factory class method) registry_name() (jaxdem.colliders.naive.NaiveSimulator class method) (jaxdem.colliders.NaiveSimulator class method) (jaxdem.domains.free.FreeDomain class method) (jaxdem.domains.FreeDomain class method) (jaxdem.domains.periodic.PeriodicDomain class method) (jaxdem.domains.PeriodicDomain class method) (jaxdem.domains.reflect.ReflectDomain class method) (jaxdem.domains.ReflectDomain class method) (jaxdem.forces.law_combiner.LawCombiner class method) (jaxdem.forces.LawCombiner class method) (jaxdem.forces.spring.SpringForce class method) (jaxdem.forces.SpringForce class method) (jaxdem.integrators.direct_euler.DirectEuler class method) (jaxdem.integrators.DirectEuler class method) (jaxdem.LawCombiner class method) (jaxdem.material_matchmakers.harmonic.HarmonicMaterialMatchmaker class method) (jaxdem.material_matchmakers.HarmonicMaterialMatchmaker class method) (jaxdem.material_matchmakers.linear.LinearMaterialMatchmaker class method) (jaxdem.material_matchmakers.LinearMaterialMatchmaker class method) (jaxdem.materials.Elastic class method) (jaxdem.materials.ElasticFriction class method) (jaxdem.materials.elasticMats.Elastic class method) (jaxdem.materials.elasticMats.ElasticFriction class method) (jaxdem.rl.actionSpaces.BoxSpace class method) (jaxdem.rl.actionSpaces.boxSpace.BoxSpace class method) (jaxdem.rl.actionSpaces.FreeSpace class method) (jaxdem.rl.actionSpaces.freeSpace.FreeSpace class method) (jaxdem.rl.actionSpaces.MaxNormSpace class method) (jaxdem.rl.actionSpaces.maxNormSpace.MaxNormSpace class method) (jaxdem.rl.environments.multi_navigator.MultiNavigator class method) (jaxdem.rl.environments.MultiNavigator class method) (jaxdem.rl.environments.single_navigator.SingleNavigator class method) (jaxdem.rl.environments.SingleNavigator class method) (jaxdem.rl.models.ActorCritic class method) (jaxdem.rl.models.LSTM.LSTMActorCritic class method) (jaxdem.rl.models.LSTMActorCritic class method) (jaxdem.rl.models.MLP.ActorCritic class method) (jaxdem.rl.models.MLP.SharedActorCritic class method) (jaxdem.rl.models.SharedActorCritic class method) (jaxdem.rl.trainers.PPOTrainer class method) (jaxdem.rl.trainers.PPOtrainer.PPOTrainer class method) (jaxdem.writers.VTKDomainWriter class method) (jaxdem.writers.vtkDomainWriter.VTKDomainWriter class method) (jaxdem.writers.VTKSpheresWriter class method) (jaxdem.writers.vtkSpheresWriter.VTKSpheresWriter class method)

required_material_properties (jaxdem.ForceModel attribute) (jaxdem.forces.ForceModel attribute) reset() (jaxdem.rl.Environment static method) (jaxdem.rl.environments.Environment static method) (jaxdem.rl.environments.multi_navigator.MultiNavigator static method) (jaxdem.rl.environments.MultiNavigator static method) (jaxdem.rl.environments.single_navigator.SingleNavigator static method) (jaxdem.rl.environments.SingleNavigator static method) (jaxdem.rl.Model method) (jaxdem.rl.models.LSTM.LSTMActorCritic method) (jaxdem.rl.models.LSTMActorCritic method) (jaxdem.rl.models.Model method) reset_if_done() (jaxdem.rl.Environment static method) (jaxdem.rl.environments.Environment static method) reset_model() (jaxdem.rl.Trainer static method) (jaxdem.rl.trainers.PPOTrainer static method) (jaxdem.rl.trainers.PPOtrainer.PPOTrainer static method) returns (jaxdem.rl.TrajectoryData attribute) reward (jaxdem.rl.TrajectoryData attribute) reward() (jaxdem.rl.Environment static method) (jaxdem.rl.environments.Environment static method) (jaxdem.rl.environments.multi_navigator.MultiNavigator static method) (jaxdem.rl.environments.MultiNavigator static method) (jaxdem.rl.environments.single_navigator.SingleNavigator static method) (jaxdem.rl.environments.SingleNavigator static method)

S

same_as() (jaxdem.rl.actionSpaces.BoxSpace method) (jaxdem.rl.actionSpaces.boxSpace.BoxSpace method) (jaxdem.rl.actionSpaces.FreeSpace method) (jaxdem.rl.actionSpaces.freeSpace.FreeSpace method) (jaxdem.rl.actionSpaces.MaxNormSpace method) (jaxdem.rl.actionSpaces.maxNormSpace.MaxNormSpace method) save() (jaxdem.CheckpointModelWriter method) (jaxdem.CheckpointWriter method) (jaxdem.VTKWriter method) (jaxdem.writers.CheckpointModelWriter method) (jaxdem.writers.checkpoints.CheckpointModelWriter method) (jaxdem.writers.checkpoints.CheckpointWriter method) (jaxdem.writers.CheckpointWriter method) (jaxdem.writers.VTKWriter method) (jaxdem.writers.vtkWriter.VTKWriter method) save_every (jaxdem.CheckpointModelWriter attribute) (jaxdem.CheckpointWriter attribute) (jaxdem.VTKWriter attribute) (jaxdem.writers.CheckpointModelWriter attribute) (jaxdem.writers.checkpoints.CheckpointModelWriter attribute) (jaxdem.writers.checkpoints.CheckpointWriter attribute) (jaxdem.writers.CheckpointWriter attribute) (jaxdem.writers.VTKWriter attribute) (jaxdem.writers.vtkWriter.VTKWriter attribute) sec2_log() (jaxdem.rl.actionSpaces.BoxSpace static method) (jaxdem.rl.actionSpaces.boxSpace.BoxSpace static method) (jaxdem.rl.actionSpaces.MaxNormSpace static method) (jaxdem.rl.actionSpaces.maxNormSpace.MaxNormSpace static method) SharedActorCritic (class in jaxdem.rl.models) (class in jaxdem.rl.models.MLP) shift() (jaxdem.Domain static method) (jaxdem.domains.Domain static method) (jaxdem.domains.free.FreeDomain static method) (jaxdem.domains.FreeDomain static method) (jaxdem.domains.periodic.PeriodicDomain static method) (jaxdem.domains.PeriodicDomain static method) (jaxdem.domains.reflect.ReflectDomain static method) (jaxdem.domains.ReflectDomain static method)

signed_angle() (in module jaxdem.utils) (in module jaxdem.utils.angles) signed_angle_x() (in module jaxdem.utils) (in module jaxdem.utils.angles) SingleNavigator (class in jaxdem.rl.environments) (class in jaxdem.rl.environments.single_navigator) species_id (jaxdem.State attribute) (jaxdem.state.State attribute) SpringForce (class in jaxdem.forces) (class in jaxdem.forces.spring) stack() (jaxdem.State static method) (jaxdem.state.State static method) State (class in jaxdem) (class in jaxdem.state) state (jaxdem.rl.Environment attribute) (jaxdem.rl.environments.Environment attribute) step() (jaxdem.Integrator static method) (jaxdem.integrators.direct_euler.DirectEuler static method) (jaxdem.integrators.DirectEuler static method) (jaxdem.integrators.Integrator static method) (jaxdem.rl.Environment static method) (jaxdem.rl.environments.Environment static method) (jaxdem.rl.environments.multi_navigator.MultiNavigator static method) (jaxdem.rl.environments.MultiNavigator static method) (jaxdem.rl.environments.single_navigator.SingleNavigator static method) (jaxdem.rl.environments.SingleNavigator static method) (jaxdem.rl.Trainer static method) (jaxdem.System static method) (jaxdem.system.System static method) step_count (jaxdem.System attribute) (jaxdem.system.System attribute) System (class in jaxdem) (class in jaxdem.system) system (jaxdem.rl.Environment attribute) (jaxdem.rl.environments.Environment attribute)

T

table (jaxdem.ForceRouter attribute) (jaxdem.forces.ForceRouter attribute) (jaxdem.forces.router.ForceRouter attribute) time (jaxdem.System attribute) (jaxdem.system.System attribute) train() (jaxdem.rl.Trainer static method) (jaxdem.rl.trainers.PPOTrainer static method) (jaxdem.rl.trainers.PPOtrainer.PPOTrainer static method) Trainer (class in jaxdem.rl) trajectory_rollout() (jaxdem.rl.Trainer static method) (jaxdem.System static method) (jaxdem.system.System static method) TrajectoryData (class in jaxdem.rl) type_name (jaxdem.colliders.naive.NaiveSimulator property) (jaxdem.colliders.NaiveSimulator property) (jaxdem.domains.free.FreeDomain property) (jaxdem.domains.FreeDomain property) (jaxdem.domains.periodic.PeriodicDomain property) (jaxdem.domains.PeriodicDomain property) (jaxdem.domains.reflect.ReflectDomain property) (jaxdem.domains.ReflectDomain property) (jaxdem.forces.law_combiner.LawCombiner property) (jaxdem.forces.LawCombiner property) (jaxdem.forces.spring.SpringForce property) (jaxdem.forces.SpringForce property) (jaxdem.integrators.direct_euler.DirectEuler property) (jaxdem.integrators.DirectEuler property) (jaxdem.LawCombiner property) (jaxdem.material_matchmakers.harmonic.HarmonicMaterialMatchmaker property) (jaxdem.material_matchmakers.HarmonicMaterialMatchmaker property) (jaxdem.material_matchmakers.linear.LinearMaterialMatchmaker property) (jaxdem.material_matchmakers.LinearMaterialMatchmaker property) (jaxdem.materials.Elastic property) (jaxdem.materials.ElasticFriction property) (jaxdem.materials.elasticMats.Elastic property) (jaxdem.materials.elasticMats.ElasticFriction property) (jaxdem.rl.actionSpaces.BoxSpace property) (jaxdem.rl.actionSpaces.boxSpace.BoxSpace property) (jaxdem.rl.actionSpaces.FreeSpace property) (jaxdem.rl.actionSpaces.freeSpace.FreeSpace property) (jaxdem.rl.actionSpaces.MaxNormSpace property) (jaxdem.rl.actionSpaces.maxNormSpace.MaxNormSpace property) (jaxdem.rl.environments.multi_navigator.MultiNavigator property) (jaxdem.rl.environments.MultiNavigator property) (jaxdem.rl.environments.single_navigator.SingleNavigator property) (jaxdem.rl.environments.SingleNavigator property) (jaxdem.rl.models.ActorCritic property) (jaxdem.rl.models.LSTM.LSTMActorCritic property) (jaxdem.rl.models.LSTMActorCritic property) (jaxdem.rl.models.MLP.ActorCritic property) (jaxdem.rl.models.MLP.SharedActorCritic property) (jaxdem.rl.models.SharedActorCritic property) (jaxdem.rl.trainers.PPOTrainer property) (jaxdem.rl.trainers.PPOtrainer.PPOTrainer property) (jaxdem.writers.VTKDomainWriter property) (jaxdem.writers.vtkDomainWriter.VTKDomainWriter property) (jaxdem.writers.VTKSpheresWriter property) (jaxdem.writers.vtkSpheresWriter.VTKSpheresWriter property)

U

unit() (in module jaxdem.utils) (in module jaxdem.utils.linalg)

unwrap() (in module jaxdem.rl) (in module jaxdem.rl.envWrappers)

V

value (jaxdem.rl.TrajectoryData attribute) vectorise_env() (in module jaxdem.rl) (in module jaxdem.rl.envWrappers) vel (jaxdem.State attribute) (jaxdem.state.State attribute) VTKBaseWriter (class in jaxdem) (class in jaxdem.writers)

VTKDomainWriter (class in jaxdem.writers) (class in jaxdem.writers.vtkDomainWriter) VTKSpheresWriter (class in jaxdem.writers) (class in jaxdem.writers.vtkSpheresWriter) VTKWriter (class in jaxdem) (class in jaxdem.writers) (class in jaxdem.writers.vtkWriter)

W

write() (jaxdem.VTKBaseWriter class method) (jaxdem.writers.VTKBaseWriter class method) (jaxdem.writers.VTKDomainWriter class method) (jaxdem.writers.vtkDomainWriter.VTKDomainWriter class method) (jaxdem.writers.VTKSpheresWriter class method) (jaxdem.writers.vtkSpheresWriter.VTKSpheresWriter class method)

writers (jaxdem.VTKWriter attribute) (jaxdem.writers.VTKWriter attribute) (jaxdem.writers.vtkWriter.VTKWriter attribute)

Y

young (jaxdem.materials.Elastic attribute) (jaxdem.materials.ElasticFriction attribute) (jaxdem.materials.elasticMats.Elastic attribute) (jaxdem.materials.elasticMats.ElasticFriction attribute)