Index A | B | C | D | E | F | G | H | I | J | L | M | N | O | P | R | S | T | V | W | Y A accel (jaxdem.state.State attribute) action_space_size (jaxdem.rl.environment.Environment attribute) (jaxdem.rl.environment.SingleNavigator attribute) add() (jaxdem.state.State static method) anchor (jaxdem.domain.Domain attribute) (jaxdem.domain.FreeDomain attribute) (jaxdem.domain.PeriodicDomain attribute) (jaxdem.domain.ReflectDomain attribute) B batch_size (jaxdem.state.State property) binary (jaxdem.writer.VTKWriter attribute) box_size (jaxdem.domain.Domain attribute) (jaxdem.domain.FreeDomain attribute) (jaxdem.domain.PeriodicDomain attribute) (jaxdem.domain.ReflectDomain attribute) C clean (jaxdem.writer.VTKWriter attribute) Collider (class in jaxdem.collider) collider (jaxdem.system.System attribute) compute_force() (jaxdem.collider.Collider static method) (jaxdem.collider.NaiveSimulator static method) compute_potential_energy() (jaxdem.collider.Collider static method) (jaxdem.collider.NaiveSimulator static method) create() (jaxdem.collider.Collider class method) (jaxdem.collider.NaiveSimulator class method) (jaxdem.domain.Domain class method) (jaxdem.domain.FreeDomain class method) (jaxdem.domain.PeriodicDomain class method) (jaxdem.domain.ReflectDomain class method) (jaxdem.factory.Factory class method) (jaxdem.forceRouter.ForceRouter class method) (jaxdem.forceRouter.LawCombiner class method) (jaxdem.forces.ForceModel class method) (jaxdem.forces.SpringForce class method) (jaxdem.integrator.DirectEuler class method) (jaxdem.integrator.Integrator class method) (jaxdem.material.Elastic class method) (jaxdem.material.ElasticFriction class method) (jaxdem.material.Material class method) (jaxdem.materialMatchmaker.HarmonicMaterialMatchmaker class method) (jaxdem.materialMatchmaker.LinearMaterialMatchmaker class method) (jaxdem.materialMatchmaker.MaterialMatchmaker class method) (jaxdem.rl.environment.Environment class method) (jaxdem.rl.environment.SingleNavigator class method) (jaxdem.state.State static method) (jaxdem.system.System static method) (jaxdem.writer.DomainWriter class method) (jaxdem.writer.SpheresWriter class method) (jaxdem.writer.VTKBaseWriter class method) D dim (jaxdem.state.State property) DirectEuler (class in jaxdem.integrator) directory (jaxdem.writer.VTKWriter attribute) displacement() (jaxdem.domain.Domain static method) (jaxdem.domain.FreeDomain static method) (jaxdem.domain.PeriodicDomain static method) (jaxdem.domain.ReflectDomain static method) Domain (class in jaxdem.domain) domain (jaxdem.system.System attribute) DomainWriter (class in jaxdem.writer) done() (jaxdem.rl.environment.Environment static method) (jaxdem.rl.environment.SingleNavigator static method) dt (jaxdem.system.System attribute) E Elastic (class in jaxdem.material) ElasticFriction (class in jaxdem.material) energy() (jaxdem.forceRouter.ForceRouter static method) (jaxdem.forceRouter.LawCombiner static method) (jaxdem.forces.ForceModel static method) (jaxdem.forces.SpringForce static method) env_params (jaxdem.rl.environment.Environment attribute) (jaxdem.rl.environment.SingleNavigator attribute) Environment (class in jaxdem.rl.environment) F Factory (class in jaxdem.factory) fixed (jaxdem.state.State attribute) force() (jaxdem.forceRouter.ForceRouter static method) (jaxdem.forceRouter.LawCombiner static method) (jaxdem.forces.ForceModel static method) (jaxdem.forces.SpringForce static method) force_model (jaxdem.system.System attribute) ForceModel (class in jaxdem.forces) ForceRouter (class in jaxdem.forceRouter) FreeDomain (class in jaxdem.domain) from_dict() (jaxdem.forceRouter.ForceRouter static method) from_materials() (jaxdem.material.MaterialTable static method) G get_effective_property() (jaxdem.materialMatchmaker.HarmonicMaterialMatchmaker static method) (jaxdem.materialMatchmaker.LinearMaterialMatchmaker static method) (jaxdem.materialMatchmaker.MaterialMatchmaker static method) grid_state() (in module jaxdem.utils) H HarmonicMaterialMatchmaker (class in jaxdem.materialMatchmaker) I ID (jaxdem.state.State attribute) info() (jaxdem.rl.environment.Environment static method) (jaxdem.rl.environment.SingleNavigator static method) initialize() (jaxdem.integrator.DirectEuler static method) (jaxdem.integrator.Integrator static method) Integrator (class in jaxdem.integrator) integrator (jaxdem.system.System attribute) is_valid (jaxdem.state.State property) J jaxdem.collider module jaxdem.domain module jaxdem.factory module jaxdem.forceRouter module jaxdem.forces module jaxdem.integrator module jaxdem.material module jaxdem.materialMatchmaker module jaxdem.rl module jaxdem.rl.environment module jaxdem.state module jaxdem.system module jaxdem.utils module jaxdem.writer module L LawCombiner (class in jaxdem.forceRouter) laws (jaxdem.forceRouter.ForceRouter attribute) (jaxdem.forceRouter.LawCombiner attribute) (jaxdem.forces.ForceModel attribute) (jaxdem.forces.SpringForce attribute) LinearMaterialMatchmaker (class in jaxdem.materialMatchmaker) M mass (jaxdem.state.State attribute) mat_id (jaxdem.state.State attribute) mat_table (jaxdem.system.System attribute) matcher (jaxdem.material.MaterialTable attribute) Material (class in jaxdem.material) MaterialMatchmaker (class in jaxdem.materialMatchmaker) MaterialTable (class in jaxdem.material) merge() (jaxdem.state.State static method) module jaxdem.collider jaxdem.domain jaxdem.factory jaxdem.forceRouter jaxdem.forces jaxdem.integrator jaxdem.material jaxdem.materialMatchmaker jaxdem.rl jaxdem.rl.environment jaxdem.state jaxdem.system jaxdem.utils jaxdem.writer mu (jaxdem.material.ElasticFriction attribute) N N (jaxdem.state.State property) NaiveSimulator (class in jaxdem.collider) O observation() (jaxdem.rl.environment.Environment static method) (jaxdem.rl.environment.SingleNavigator static method) observation_space_size (jaxdem.rl.environment.Environment attribute) (jaxdem.rl.environment.SingleNavigator attribute) P pair (jaxdem.material.MaterialTable attribute) periodic (jaxdem.domain.Domain attribute) (jaxdem.domain.FreeDomain attribute) (jaxdem.domain.PeriodicDomain attribute) (jaxdem.domain.ReflectDomain attribute) PeriodicDomain (class in jaxdem.domain) poisson (jaxdem.material.Elastic attribute) (jaxdem.material.ElasticFriction attribute) pos (jaxdem.state.State attribute) props (jaxdem.material.MaterialTable attribute) R rad (jaxdem.state.State attribute) random_state() (in module jaxdem.utils) ReflectDomain (class in jaxdem.domain) register() (jaxdem.collider.Collider class method) (jaxdem.collider.NaiveSimulator class method) (jaxdem.domain.Domain class method) (jaxdem.domain.FreeDomain class method) (jaxdem.domain.PeriodicDomain class method) (jaxdem.domain.ReflectDomain class method) (jaxdem.factory.Factory class method) (jaxdem.forceRouter.ForceRouter class method) (jaxdem.forceRouter.LawCombiner class method) (jaxdem.forces.ForceModel class method) (jaxdem.forces.SpringForce class method) (jaxdem.integrator.DirectEuler class method) (jaxdem.integrator.Integrator class method) (jaxdem.material.Elastic class method) (jaxdem.material.ElasticFriction class method) (jaxdem.material.Material class method) (jaxdem.materialMatchmaker.HarmonicMaterialMatchmaker class method) (jaxdem.materialMatchmaker.LinearMaterialMatchmaker class method) (jaxdem.materialMatchmaker.MaterialMatchmaker class method) (jaxdem.rl.environment.Environment class method) (jaxdem.rl.environment.SingleNavigator class method) (jaxdem.writer.DomainWriter class method) (jaxdem.writer.SpheresWriter class method) (jaxdem.writer.VTKBaseWriter class method) required_material_properties (jaxdem.forceRouter.ForceRouter attribute) (jaxdem.forceRouter.LawCombiner attribute) (jaxdem.forces.ForceModel attribute) (jaxdem.forces.SpringForce attribute) reset() (jaxdem.rl.environment.Environment static method) (jaxdem.rl.environment.SingleNavigator static method) reward() (jaxdem.rl.environment.Environment static method) (jaxdem.rl.environment.SingleNavigator static method) S save() (jaxdem.writer.VTKWriter method) shift() (jaxdem.domain.Domain static method) (jaxdem.domain.FreeDomain static method) (jaxdem.domain.PeriodicDomain static method) (jaxdem.domain.ReflectDomain static method) SingleNavigator (class in jaxdem.rl.environment) species_id (jaxdem.state.State attribute) SpheresWriter (class in jaxdem.writer) SpringForce (class in jaxdem.forces) stack() (jaxdem.state.State static method) State (class in jaxdem.state) state (jaxdem.rl.environment.Environment attribute) (jaxdem.rl.environment.SingleNavigator attribute) step() (jaxdem.integrator.DirectEuler static method) (jaxdem.integrator.Integrator static method) (jaxdem.rl.environment.Environment static method) (jaxdem.rl.environment.SingleNavigator static method) (jaxdem.system.System static method) step_count (jaxdem.system.System attribute) System (class in jaxdem.system) system (jaxdem.rl.environment.Environment attribute) (jaxdem.rl.environment.SingleNavigator attribute) T table (jaxdem.forceRouter.ForceRouter attribute) trajectory_rollout() (jaxdem.system.System static method) V vel (jaxdem.state.State attribute) VTKBaseWriter (class in jaxdem.writer) VTKWriter (class in jaxdem.writer) W write() (jaxdem.writer.DomainWriter class method) (jaxdem.writer.SpheresWriter class method) (jaxdem.writer.VTKBaseWriter class method) writers (jaxdem.writer.VTKWriter attribute) Y young (jaxdem.material.Elastic attribute) (jaxdem.material.ElasticFriction attribute)