gmlp-flax (pypi): Flax implementation of gMLP from Pay Attention to MLPs
Keywords: deep-learning|flax|jax|machine-learning|neural-networks|pypi|pypi-package|python