Don't know if its worth looking at, but you can look at my code: https://github.com/Vrroom/vectorrvnn :)