GitHub repository: EdanToledo/flashbax
Primary language: Python
Description: ⚡ Flashbax: Accelerated Replay Buffers in JAX