GitHub repository: keraJLi/rejax
Primary language: Python
Description: Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!