A Pythonic, GPU Friendly implementation of AGILEPack