src/
├── config/
│ └── default.yaml # Configuration file (provided)
├── lib/
│ ├── physics.py # Physics engine (implement this)
│ └── models.py # Neural networks (implement this)
├── helpers/
│ ├── utils.py # Data loading utilities (provided)
│ └── config.py # Config handling (provided)
└── main.py # Training script (partially provided)
Install requirements:
pip install -r requirements.txt