7 projects
cyreal
Jittable data loading utilities for JAX.
popgym-arcade
POMDP Arcade Environments on the GPU
memax
Deep memory and sequence modeling in JAX
stable-gymnax
(Stable) JAX implementations of OpenAI's gym environments
popgym
A collection of partially-observable procedural gym environments
genagg
A learnable, generalised aggregation module for pytorch that parametrises the space of aggregation functions.
graph-conv-memory
Graph convolutional memory for reinforcement learning
