I do not recommend using a neutral network as it is for more effecient to use simpler PID controllers, odometry, motion profiling etc. If you really wanted to use reinforcement learning you will probably have to use c++ for performance reasons. You could turn looking at this library: Copilot: An ML Library for PROS V5y