Examples

• Intro to JaxDEM Reinforcement Learning.
  • Imports
  • Environment
  • Model
  • Trainer (PPO)
  • Training
  • Testing the New Policy
• Driving Environments with a Custom Policy.
  • Imports
  • Variables
  • The Policy
  • Model and Environment
  • Environment Vectorization
  • Driving the Environment
  • Saving Data
• Minimization of bidisperse spheres (or disks).
  • Imports
  • Parameters
  • Run the Minimization for Multiple Systems
  • Run the Minimization for a Single System
• Jamming of bidisperse spheres (or disks).
  • Imports
  • Parameters
  • Run the Jamming Algorithm for Multiple Systems
  • Run the Jamming Algorithm for a Single System
• Pseudo-log trajectory rollout.
• Imports
• Parameters
• Run dynamics while imposing a sine-wave compression
• Vicsek Model