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