DeepMind's Gated Linear Networks in NumPy, PyTorch, TensorFlow and JAXgithub.com/aiwabdn 6aiwabdn6y1 comment