Offline Learning for Stochastic Multi-Agent Planning in Autonomous Driving