dm-haiku topic

JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"

evgenii-nikishin

Experiment with diffusion models that you can run on your local jupyter instances

varun-ml

Spyx: Spiking Neural Networks in JAX

kmheckel

Codes for "Efficient Offline Policy Optimization with a Learned Model", ICLR2023

sail-sg