a simple and customizable library for training RL and imitation learning models using flax !!! this library is still in development and is not yet ready for use !!!
pip install irlaxRL algorithms
- DQN
- DQN
- Double DQN
- QR-DQN
- PPO
- TD3
Imitation learning algorithms
- BC
- DAgger
Replay Buffers
- Uniform Replay Buffer
- Prioritized Replay Buffer
- python
- concurrent rust implementation
callbacks
- learning curve plotter