Index

A

action (jaxdem.rl.TrajectoryData attribute) action_space_shape (jaxdem.rl.Environment property) (jaxdem.rl.environments.Environment 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) action_space_size (jaxdem.rl.Environment property) (jaxdem.rl.environments.Environment 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) ActionSpace (class in jaxdem.rl) (class in jaxdem.rl.actionSpaces) actor_mu (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_sigma (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) add_clump() (jaxdem.State static method) (jaxdem.state.State static method) add_force() (jaxdem.ForceManager static method) (jaxdem.forces.force_manager.ForceManager static method) (jaxdem.forces.ForceManager static method) add_force_at() (jaxdem.ForceManager static method) (jaxdem.forces.force_manager.ForceManager static method) (jaxdem.forces.ForceManager static method)

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) allocate_counts() (in module jaxdem.utils.dispersity) alpha (jaxdem.minimizers.fire.LinearFIRE attribute) (jaxdem.minimizers.fire.RotationFIRE attribute) (jaxdem.minimizers.LinearFIRE attribute) (jaxdem.minimizers.RotationFIRE attribute) alpha_init (jaxdem.minimizers.fire.LinearFIRE attribute) (jaxdem.minimizers.fire.RotationFIRE attribute) (jaxdem.minimizers.LinearFIRE attribute) (jaxdem.minimizers.RotationFIRE 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) angVel (jaxdem.State attribute) (jaxdem.state.State attribute) anneal_importance_sampling_beta (jaxdem.rl.trainers.PPOTrainer attribute) (jaxdem.rl.trainers.PPOtrainer.PPOTrainer attribute) apply() (jaxdem.Domain static method) (jaxdem.domains.Domain static method) (jaxdem.domains.free.FreeDomain static method) (jaxdem.domains.FreeDomain static method) (jaxdem.domains.reflect.ReflectDomain static method) (jaxdem.domains.reflect_sphere.ReflectSphereDomain static method) (jaxdem.domains.ReflectDomain static method) (jaxdem.domains.ReflectSphereDomain static method) (jaxdem.ForceManager static method) (jaxdem.forces.force_manager.ForceManager static method) (jaxdem.forces.ForceManager static method) attempt_couple (jaxdem.minimizers.fire.LinearFIRE attribute) (jaxdem.minimizers.fire.RotationFIRE attribute) (jaxdem.minimizers.LinearFIRE attribute) (jaxdem.minimizers.RotationFIRE 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) bisection_jam() (in module jaxdem.utils.jamming) 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) cell_size (jaxdem.colliders.cell_list.CellList attribute) (jaxdem.colliders.cell_list.DynamicCellList attribute) (jaxdem.colliders.cell_list.MaterializedCellList attribute) (jaxdem.colliders.cell_list.NeighborList attribute) (jaxdem.colliders.CellList attribute) (jaxdem.colliders.DynamicCellList attribute) (jaxdem.colliders.MaterializedCellList attribute) (jaxdem.colliders.NeighborList attribute) CellList (class in jaxdem.colliders) (class in jaxdem.colliders.cell_list) 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_clump_properties() (in module jaxdem.utils) (in module jaxdem.utils.clumps) compute_force() (jaxdem.Collider static method) (jaxdem.colliders.cell_list.CellList static method) (jaxdem.colliders.cell_list.DynamicCellList static method) (jaxdem.colliders.cell_list.MaterializedCellList static method) (jaxdem.colliders.cell_list.NeighborList static method) (jaxdem.colliders.CellList static method) (jaxdem.colliders.Collider static method) (jaxdem.colliders.DynamicCellList static method) (jaxdem.colliders.MaterializedCellList static method) (jaxdem.colliders.naive.NaiveSimulator static method) (jaxdem.colliders.NaiveSimulator static method) (jaxdem.colliders.NeighborList static method) (jaxdem.colliders.sweep_and_prune.SweeAPrune static method)

compute_hash() (in module jaxdem.colliders.sweep_and_prune) compute_potential_energy() (jaxdem.Collider static method) (jaxdem.colliders.cell_list.CellList static method) (jaxdem.colliders.cell_list.DynamicCellList static method) (jaxdem.colliders.cell_list.MaterializedCellList static method) (jaxdem.colliders.cell_list.NeighborList static method) (jaxdem.colliders.CellList static method) (jaxdem.colliders.Collider static method) (jaxdem.colliders.DynamicCellList static method) (jaxdem.colliders.MaterializedCellList static method) (jaxdem.colliders.naive.NaiveSimulator static method) (jaxdem.colliders.NaiveSimulator static method) (jaxdem.colliders.NeighborList static method) (jaxdem.colliders.sweep_and_prune.SweeAPrune static method) compute_virtual_shift() (in module jaxdem.colliders.sweep_and_prune) conj() (jaxdem.utils.Quaternion static method) (jaxdem.utils.quaternion.Quaternion static method) coupled (jaxdem.minimizers.fire.LinearFIRE attribute) (jaxdem.minimizers.fire.RotationFIRE attribute) (jaxdem.minimizers.LinearFIRE attribute) (jaxdem.minimizers.RotationFIRE attribute) Create() (jaxdem.colliders.cell_list.CellList class method) (jaxdem.colliders.cell_list.DynamicCellList class method) (jaxdem.colliders.cell_list.MaterializedCellList class method) (jaxdem.colliders.cell_list.NeighborList class method) (jaxdem.colliders.CellList class method) (jaxdem.colliders.DynamicCellList class method) (jaxdem.colliders.MaterializedCellList class method) (jaxdem.colliders.NeighborList class method) (jaxdem.Domain class method) (jaxdem.domains.Domain class method) (jaxdem.domains.reflect.ReflectDomain class method) (jaxdem.domains.ReflectDomain class method) create() (jaxdem.Factory class method) (jaxdem.factory.Factory class method) (jaxdem.ForceManager static method) (jaxdem.forces.force_manager.ForceManager static method) (jaxdem.forces.ForceManager static method) Create() (jaxdem.minimizers.fire.LinearFIRE class method) (jaxdem.minimizers.fire.RotationFIRE class method) (jaxdem.minimizers.gradient_descent.LinearGradientDescent class method) (jaxdem.minimizers.gradient_descent.RotationGradientDescent class method) (jaxdem.minimizers.LinearFIRE class method) (jaxdem.minimizers.LinearGradientDescent class method) (jaxdem.minimizers.RotationFIRE class method) (jaxdem.minimizers.RotationGradientDescent 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.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) (jaxdem.utils.Quaternion static method) (jaxdem.utils.quaternion.Quaternion 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) cross() (in module jaxdem.utils.linalg) cross_3X3D_1X2D() (in module jaxdem.utils) (in module jaxdem.utils.linalg)

D

decode_callable() (in module jaxdem.utils) (in module jaxdem.utils.serialization) DeepOnetActorCritic (class in jaxdem.rl.models) (class in jaxdem.rl.models.DeepONet) density (jaxdem.Material attribute) (jaxdem.materials.Material attribute) did_buffer_overflow (jaxdem.colliders.cell_list.NeighborList attribute) (jaxdem.colliders.NeighborList attribute) 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.periodic.PeriodicDomain static method) (jaxdem.domains.PeriodicDomain 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.minimizers.fire.LinearFIRE attribute) (jaxdem.minimizers.fire.RotationFIRE attribute) (jaxdem.minimizers.LinearFIRE attribute) (jaxdem.minimizers.RotationFIRE attribute) (jaxdem.System attribute) (jaxdem.system.System attribute) dt_max (jaxdem.minimizers.fire.LinearFIRE attribute) (jaxdem.minimizers.fire.RotationFIRE attribute) (jaxdem.minimizers.LinearFIRE attribute) (jaxdem.minimizers.RotationFIRE attribute) dt_max_scale (jaxdem.minimizers.fire.LinearFIRE attribute) (jaxdem.minimizers.fire.RotationFIRE attribute) (jaxdem.minimizers.LinearFIRE attribute) (jaxdem.minimizers.RotationFIRE attribute) dt_min (jaxdem.minimizers.fire.LinearFIRE attribute) (jaxdem.minimizers.fire.RotationFIRE attribute) (jaxdem.minimizers.LinearFIRE attribute) (jaxdem.minimizers.RotationFIRE attribute) dt_min_scale (jaxdem.minimizers.fire.LinearFIRE attribute) (jaxdem.minimizers.fire.RotationFIRE attribute) (jaxdem.minimizers.LinearFIRE attribute) (jaxdem.minimizers.RotationFIRE attribute) dt_reverse (jaxdem.minimizers.fire.LinearFIRE attribute) (jaxdem.minimizers.fire.RotationFIRE attribute) (jaxdem.minimizers.LinearFIRE attribute) (jaxdem.minimizers.RotationFIRE attribute) dtype (jaxdem.colliders.sweep_and_prune.RefProxy property) DynamicCellList (class in jaxdem.colliders) (class in jaxdem.colliders.cell_list)

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) env_step() (in module jaxdem.utils) (in module jaxdem.utils.environment) env_trajectory_rollout() (in module jaxdem.utils) (in module jaxdem.utils.environment) 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) external_force (jaxdem.ForceManager attribute) (jaxdem.forces.force_manager.ForceManager attribute) (jaxdem.forces.ForceManager attribute) external_torque (jaxdem.ForceManager attribute) (jaxdem.forces.force_manager.ForceManager attribute) (jaxdem.forces.ForceManager attribute)

F

f_alpha (jaxdem.minimizers.fire.LinearFIRE attribute) (jaxdem.minimizers.fire.RotationFIRE attribute) (jaxdem.minimizers.LinearFIRE attribute) (jaxdem.minimizers.RotationFIRE attribute) f_dec (jaxdem.minimizers.fire.LinearFIRE attribute) (jaxdem.minimizers.fire.RotationFIRE attribute) (jaxdem.minimizers.LinearFIRE attribute) (jaxdem.minimizers.RotationFIRE attribute) f_inc (jaxdem.minimizers.fire.LinearFIRE attribute) (jaxdem.minimizers.fire.RotationFIRE attribute) (jaxdem.minimizers.LinearFIRE attribute) (jaxdem.minimizers.RotationFIRE attribute) Factory (class in jaxdem) (class in jaxdem.factory) fixed (jaxdem.State attribute) (jaxdem.state.State attribute) force (jaxdem.State attribute) (jaxdem.state.State attribute) force() (in module jaxdem.colliders.sweep_and_prune) (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_functions (jaxdem.ForceManager attribute) (jaxdem.forces.force_manager.ForceManager attribute) (jaxdem.forces.ForceManager attribute)

force_manager (jaxdem.System attribute) (jaxdem.system.System attribute) force_model (jaxdem.System attribute) (jaxdem.system.System attribute) ForceManager (class in jaxdem) (class in jaxdem.forces) (class in jaxdem.forces.force_manager) 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) get_polydisperse_radii() (in module jaxdem.utils.dispersity) get_real_pos() (in module jaxdem.state) graphdef (jaxdem.rl.Trainer attribute)

graphstate (jaxdem.rl.Trainer attribute) gravity (jaxdem.ForceManager attribute) (jaxdem.forces.force_manager.ForceManager attribute) (jaxdem.forces.ForceManager attribute) grid_dims (jaxdem.colliders.cell_list.MaterializedCellList attribute) (jaxdem.colliders.cell_list.NeighborList attribute) (jaxdem.colliders.MaterializedCellList attribute) (jaxdem.colliders.NeighborList 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) idx (jaxdem.colliders.cell_list.NeighborList attribute) (jaxdem.colliders.NeighborList 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) inertia (jaxdem.State attribute) (jaxdem.state.State attribute) info() (jaxdem.rl.Environment static method) (jaxdem.rl.environments.Environment static method) initialize() (jaxdem.Integrator static method) (jaxdem.minimizers.fire.LinearFIRE static method) (jaxdem.minimizers.fire.RotationFIRE static method) (jaxdem.minimizers.LinearFIRE static method) (jaxdem.minimizers.RotationFIRE static method)

Integrator (class in jaxdem) inv() (jaxdem.utils.Quaternion static method) (jaxdem.utils.quaternion.Quaternion static method) 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_master (jaxdem.minimizers.fire.LinearFIRE attribute) (jaxdem.minimizers.fire.RotationFIRE attribute) (jaxdem.minimizers.LinearFIRE attribute) (jaxdem.minimizers.RotationFIRE attribute) 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.cell_list module jaxdem.colliders.naive module jaxdem.colliders.sweep_and_prune module jaxdem.domains module jaxdem.domains.free module jaxdem.domains.periodic module jaxdem.domains.reflect module jaxdem.domains.reflect_sphere module jaxdem.factory module jaxdem.forces module jaxdem.forces.force_manager module jaxdem.forces.law_combiner module jaxdem.forces.router module jaxdem.forces.spring module jaxdem.integrators module jaxdem.integrators.direct_euler module jaxdem.integrators.spiral module jaxdem.integrators.velocity_verlet module jaxdem.integrators.velocity_verlet_spiral 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.minimizers module jaxdem.minimizers.fire module jaxdem.minimizers.gradient_descent module jaxdem.minimizers.routines 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.DeepONet 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.clumps module jaxdem.utils.dispersity module jaxdem.utils.environment module jaxdem.utils.gridState module jaxdem.utils.h5 module jaxdem.utils.jamming module jaxdem.utils.linalg module jaxdem.utils.quaternion module jaxdem.utils.randomSphereConfiguration 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) learning_rate (jaxdem.minimizers.gradient_descent.LinearGradientDescent attribute) (jaxdem.minimizers.gradient_descent.RotationGradientDescent attribute) (jaxdem.minimizers.LinearGradientDescent attribute) (jaxdem.minimizers.RotationGradientDescent attribute) lidar() (in module jaxdem.utils) (in module jaxdem.utils.environment) linear_integrator (jaxdem.System attribute) (jaxdem.system.System attribute) LinearFIRE (class in jaxdem.minimizers) (class in jaxdem.minimizers.fire) LinearGradientDescent (class in jaxdem.minimizers) (class in jaxdem.minimizers.gradient_descent)

LinearIntegrator (class in jaxdem) (class in jaxdem.integrators) LinearMaterialMatchmaker (class in jaxdem.material_matchmakers) (class in jaxdem.material_matchmakers.linear) LinearMinimizer (class in jaxdem) (class in jaxdem.minimizers) load() (in module jaxdem.utils.h5) (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) 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) MaterializedCellList (class in jaxdem.colliders) (class in jaxdem.colliders.cell_list) MaterialMatchmaker (class in jaxdem) (class in jaxdem.material_matchmakers) MaterialTable (class in jaxdem) (class in jaxdem.materials) (class in jaxdem.materials.materialTable) max_neighbors (jaxdem.colliders.cell_list.NeighborList attribute) (jaxdem.colliders.NeighborList attribute) max_num_agents (jaxdem.rl.Environment property) (jaxdem.rl.environments.Environment property) max_occupancy (jaxdem.colliders.cell_list.CellList attribute) (jaxdem.colliders.cell_list.DynamicCellList attribute) (jaxdem.colliders.cell_list.MaterializedCellList attribute) (jaxdem.colliders.cell_list.NeighborList attribute) (jaxdem.colliders.CellList attribute) (jaxdem.colliders.DynamicCellList attribute) (jaxdem.colliders.MaterializedCellList attribute) (jaxdem.colliders.NeighborList 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.DeepONet.DeepOnetActorCritic property) (jaxdem.rl.models.DeepOnetActorCritic 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) minimize() (in module jaxdem.minimizers) (in module jaxdem.minimizers.routines) Minimizer (class in jaxdem) (class in jaxdem.minimizers) Model (class in jaxdem.rl) (class in jaxdem.rl.models) model (jaxdem.rl.Trainer property) module jaxdem jaxdem.colliders jaxdem.colliders.cell_list jaxdem.colliders.naive jaxdem.colliders.sweep_and_prune jaxdem.domains jaxdem.domains.free jaxdem.domains.periodic jaxdem.domains.reflect jaxdem.domains.reflect_sphere jaxdem.factory jaxdem.forces jaxdem.forces.force_manager jaxdem.forces.law_combiner jaxdem.forces.router jaxdem.forces.spring jaxdem.integrators jaxdem.integrators.direct_euler jaxdem.integrators.spiral jaxdem.integrators.velocity_verlet jaxdem.integrators.velocity_verlet_spiral jaxdem.material_matchmakers jaxdem.material_matchmakers.harmonic jaxdem.material_matchmakers.linear jaxdem.materials jaxdem.materials.elasticMats jaxdem.materials.materialTable jaxdem.minimizers jaxdem.minimizers.fire jaxdem.minimizers.gradient_descent jaxdem.minimizers.routines 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.DeepONet 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.clumps jaxdem.utils.dispersity jaxdem.utils.environment jaxdem.utils.gridState jaxdem.utils.h5 jaxdem.utils.jamming jaxdem.utils.linalg jaxdem.utils.quaternion jaxdem.utils.randomSphereConfiguration 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) N_bad (jaxdem.minimizers.fire.LinearFIRE attribute) (jaxdem.minimizers.fire.RotationFIRE attribute) (jaxdem.minimizers.LinearFIRE attribute) (jaxdem.minimizers.RotationFIRE attribute) N_bad_max (jaxdem.minimizers.fire.LinearFIRE attribute) (jaxdem.minimizers.fire.RotationFIRE attribute) (jaxdem.minimizers.LinearFIRE attribute) (jaxdem.minimizers.RotationFIRE attribute) N_good (jaxdem.minimizers.fire.LinearFIRE attribute) (jaxdem.minimizers.fire.RotationFIRE attribute) (jaxdem.minimizers.LinearFIRE attribute) (jaxdem.minimizers.RotationFIRE attribute) n_lidar_rays (jaxdem.rl.environments.multi_navigator.MultiNavigator attribute) (jaxdem.rl.environments.MultiNavigator attribute) N_min (jaxdem.minimizers.fire.LinearFIRE attribute) (jaxdem.minimizers.fire.RotationFIRE attribute) (jaxdem.minimizers.LinearFIRE attribute) (jaxdem.minimizers.RotationFIRE attribute) NaiveSimulator (class in jaxdem.colliders) (class in jaxdem.colliders.naive)

ndim (jaxdem.colliders.sweep_and_prune.RefProxy property) neighbor_mask (jaxdem.colliders.cell_list.CellList attribute) (jaxdem.colliders.cell_list.DynamicCellList attribute) (jaxdem.colliders.cell_list.MaterializedCellList attribute) (jaxdem.colliders.cell_list.NeighborList attribute) (jaxdem.colliders.CellList attribute) (jaxdem.colliders.DynamicCellList attribute) (jaxdem.colliders.MaterializedCellList attribute) (jaxdem.colliders.NeighborList attribute) NeighborList (class in jaxdem.colliders) (class in jaxdem.colliders.cell_list) network (jaxdem.rl.models.MLP.SharedActorCritic attribute) (jaxdem.rl.models.SharedActorCritic attribute) num_cells (jaxdem.colliders.cell_list.MaterializedCellList attribute) (jaxdem.colliders.MaterializedCellList attribute) num_envs (jaxdem.rl.Environment property) (jaxdem.rl.environments.Environment property) 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_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 property) (jaxdem.rl.environments.Environment 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) one_epoch() (jaxdem.rl.trainers.PPOTrainer static method) (jaxdem.rl.trainers.PPOtrainer.PPOTrainer static method)

P

pad_to_power2() (in module jaxdem.colliders.sweep_and_prune) padd() (in module jaxdem.colliders.sweep_and_prune) pair (jaxdem.materials.MaterialTable attribute) (jaxdem.materials.materialTable.MaterialTable attribute) (jaxdem.MaterialTable attribute) periodic (jaxdem.Domain property) (jaxdem.domains.Domain property) (jaxdem.domains.periodic.PeriodicDomain property) (jaxdem.domains.PeriodicDomain property) 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 property) (jaxdem.state.State property)

pos_c (jaxdem.State attribute) (jaxdem.state.State attribute) pos_p (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) prev_pos (jaxdem.colliders.cell_list.NeighborList attribute) (jaxdem.colliders.NeighborList attribute) props (jaxdem.materials.MaterialTable attribute) (jaxdem.materials.materialTable.MaterialTable attribute) (jaxdem.MaterialTable attribute)

Q

q (jaxdem.State attribute) (jaxdem.state.State attribute)

Quaternion (class in jaxdem.utils) (class in jaxdem.utils.quaternion)

R

rad (jaxdem.State attribute) (jaxdem.state.State attribute) random_sphere_configuration() (in module jaxdem.utils) (in module jaxdem.utils.randomSphereConfiguration) random_state() (in module jaxdem.utils) (in module jaxdem.utils.randomState) ratio (jaxdem.rl.TrajectoryData attribute) ReflectDomain (class in jaxdem.domains) (class in jaxdem.domains.reflect) ReflectSphereDomain (class in jaxdem.domains) (class in jaxdem.domains.reflect_sphere) RefProxy (class in jaxdem.colliders.sweep_and_prune) register() (jaxdem.Factory class method) (jaxdem.factory.Factory class method) registry_name() (jaxdem.Factory class method) (jaxdem.factory.Factory class method) required_material_properties (jaxdem.ForceModel property) (jaxdem.forces.ForceModel property) (jaxdem.forces.spring.SpringForce property) (jaxdem.forces.SpringForce property) 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) restitution_coefficient (jaxdem.domains.reflect.ReflectDomain attribute) (jaxdem.domains.ReflectDomain 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) rotate() (jaxdem.utils.Quaternion static method) (jaxdem.utils.quaternion.Quaternion static method) rotate_back() (jaxdem.utils.Quaternion static method) (jaxdem.utils.quaternion.Quaternion static method) rotation_integrator (jaxdem.System attribute) (jaxdem.system.System attribute) RotationFIRE (class in jaxdem.minimizers) (class in jaxdem.minimizers.fire) RotationGradientDescent (class in jaxdem.minimizers) (class in jaxdem.minimizers.gradient_descent) RotationIntegrator (class in jaxdem) (class in jaxdem.integrators) RotationMinimizer (class in jaxdem) (class in jaxdem.minimizers)

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) sap_kernel_full() (in module jaxdem.colliders.sweep_and_prune) save() (in module jaxdem.utils.h5) (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) shape (jaxdem.colliders.sweep_and_prune.RefProxy property) (jaxdem.State property) (jaxdem.state.State property) SharedActorCritic (class in jaxdem.rl.models) (class in jaxdem.rl.models.MLP) shift() (jaxdem.Domain static method) (jaxdem.domains.Domain static method) (jaxdem.domains.periodic.PeriodicDomain static method) (jaxdem.domains.PeriodicDomain 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) sort() (in module jaxdem.colliders.sweep_and_prune) species_id (jaxdem.State attribute) (jaxdem.state.State attribute) Spiral (class in jaxdem.integrators) (class in jaxdem.integrators.spiral) SpringForce (class in jaxdem.forces) (class in jaxdem.forces.spring) stack() (jaxdem.State static method) (jaxdem.state.State static method) (jaxdem.System static method) (jaxdem.system.System static method)

State (class in jaxdem) (class in jaxdem.state) state (jaxdem.rl.Environment attribute) (jaxdem.rl.environments.Environment attribute) step() (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_after_force() (jaxdem.Integrator static method) (jaxdem.integrators.direct_euler.DirectEuler static method) (jaxdem.integrators.DirectEuler static method) (jaxdem.integrators.Spiral static method) (jaxdem.integrators.spiral.Spiral static method) (jaxdem.integrators.velocity_verlet.VelocityVerlet static method) (jaxdem.integrators.velocity_verlet_spiral.VelocityVerletSpiral static method) (jaxdem.integrators.VelocityVerlet static method) (jaxdem.integrators.VelocityVerletSpiral static method) (jaxdem.minimizers.fire.LinearFIRE static method) (jaxdem.minimizers.fire.RotationFIRE static method) (jaxdem.minimizers.gradient_descent.LinearGradientDescent static method) (jaxdem.minimizers.gradient_descent.RotationGradientDescent static method) (jaxdem.minimizers.LinearFIRE static method) (jaxdem.minimizers.LinearGradientDescent static method) (jaxdem.minimizers.RotationFIRE static method) (jaxdem.minimizers.RotationGradientDescent static method) step_before_force() (jaxdem.Integrator static method) (jaxdem.integrators.velocity_verlet.VelocityVerlet static method) (jaxdem.integrators.velocity_verlet_spiral.VelocityVerletSpiral static method) (jaxdem.integrators.VelocityVerlet static method) (jaxdem.integrators.VelocityVerletSpiral static method) (jaxdem.minimizers.fire.LinearFIRE static method) (jaxdem.minimizers.fire.RotationFIRE static method) (jaxdem.minimizers.LinearFIRE static method) (jaxdem.minimizers.RotationFIRE static method) step_count (jaxdem.System attribute) (jaxdem.system.System attribute) stop_at_epoch (jaxdem.rl.trainers.PPOTrainer attribute) (jaxdem.rl.trainers.PPOtrainer.PPOTrainer attribute) strides (jaxdem.colliders.cell_list.MaterializedCellList attribute) (jaxdem.colliders.cell_list.NeighborList attribute) (jaxdem.colliders.MaterializedCellList attribute) (jaxdem.colliders.NeighborList attribute) SweeAPrune (class in jaxdem.colliders.sweep_and_prune) 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) torque (jaxdem.State attribute) (jaxdem.state.State 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.Factory property) (jaxdem.factory.Factory property)

U

unit() (in module jaxdem.utils) (in module jaxdem.utils.linalg) (jaxdem.utils.Quaternion static method) (jaxdem.utils.quaternion.Quaternion static method)

unwrap() (in module jaxdem.rl) (in module jaxdem.rl.envWrappers) update_threshold (jaxdem.colliders.cell_list.NeighborList attribute) (jaxdem.colliders.NeighborList attribute)

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) velocity_scale (jaxdem.minimizers.fire.LinearFIRE attribute) (jaxdem.minimizers.fire.RotationFIRE attribute) (jaxdem.minimizers.LinearFIRE attribute) (jaxdem.minimizers.RotationFIRE attribute) VelocityVerlet (class in jaxdem.integrators) (class in jaxdem.integrators.velocity_verlet) VelocityVerletSpiral (class in jaxdem.integrators) (class in jaxdem.integrators.velocity_verlet_spiral)

volume (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

w (jaxdem.utils.Quaternion attribute) (jaxdem.utils.quaternion.Quaternion attribute) 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)

X

xyz (jaxdem.utils.Quaternion attribute) (jaxdem.utils.quaternion.Quaternion attribute)

Y

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