Ray Summit

A 24x Speedup for Reinforcement Learning with RLlib + Ray

Tuesday, June 22, 7:25PM UTC

Raoul Khouri, Quantitative Software Engineer, Two Sigma

View Slides >>>

Training a reinforcement learning (RL) agent is compute intensive. Under classical deep learning assumptions bigger and better GPUs reduce training time. However, for RL, bigger and better GPUs do not always lead to reduced training time. In practice, RL can require millions of samples from a relatively slow and CPU-only environment leading to a bottleneck in training that GPUs do not solve. Empirically, we find that training agents with RLlib removes this bottleneck because its Ray integration allows scaling to many CPUs across a cluster of commodity machines. This talk details how such scaling can cut training wall-time down by orders of magnitude.

Speakers

Raoul Khouri

Raoul Khouri

Quantitative Software Engineer, Two Sigma

Raoul is a Quantitative Software Engineer at Two Sigma. He currently works in the AI Core team where he consults and collaborates on AI research with other teams around Two Sigma. Prior to his time at Two Sigma, Raoul received a B.S. and MEng. in compute science concentrated in AI from MIT.