Links for optax-swag