DeepMind's Gated Linear Networks in NumPy, PyTorch, TensorFlow and JAXgithub.com/aiwabdn6 pointsaiwabdn6 years ago