⚡ Flashbax: Accelerated Replay Buffers in JAX
Compact writing utility for storing and retrieving messages
