Stars
Unified Implementations of Offline Reinforcement Learning Algorithms
ViT Prisma is a mechanistic interpretability library for Vision and Video Transformers (ViTs).
Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!
Drop-in environment replacements that make your RL algorithm train faster.