JAX

Rainbow: The Colorful Evolution of Deep Q-Networks 🌈 featured image

Rainbow: The Colorful Evolution of Deep Q-Networks 🌈

Published on Towards Data Science

avatar
Ryan Pégoud
Read more
A Practical Guide to Proximal Policy Optimization in JAX 📈 featured image

A Practical Guide to Proximal Policy Optimization in JAX 📈

Published on Towards Data Science

avatar
Ryan Pégoud
Read more
A Gentle Introduction to Deep Reinforcement Learning in JAX 🕹️ featured image

A Gentle Introduction to Deep Reinforcement Learning in JAX 🕹️

Published on Towards Data Science

avatar
Ryan Pégoud
Read more
Implementing a Transformer Encoder from Scratch with JAX and Haiku 🤖 featured image

Implementing a Transformer Encoder from Scratch with JAX and Haiku 🤖

Published on Towards Data Science

avatar
Ryan Pégoud
Read more