This blog post details our interpretation of Gated Linear Networks, a new family of neural networks proposed by researchers at DeepMind. Implementations in JAX, NumPy, PyTorch and Tensorflow are here: https://github.com/aiwabdn/pygln
Main reason for now is that the cosine similarity used as context function is not suitable for natural images like those in Imagenet. We are experimenting some extensions based on convolutions for such use cases.