All modules for which code is available

• 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.actionSpaces
• jaxdem.rl.actionSpaces.boxSpace
• jaxdem.rl.actionSpaces.freeSpace
• jaxdem.rl.actionSpaces.maxNormSpace
• jaxdem.rl.envWrappers
• jaxdem.rl.environments
• jaxdem.rl.environments.multi_navigator
• jaxdem.rl.environments.single_navigator
• jaxdem.rl.models
• jaxdem.rl.models.LSTM
• jaxdem.rl.models.MLP
• jaxdem.rl.trainers
• jaxdem.rl.trainers.PPOtrainer
• jaxdem.state
• jaxdem.system
• 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