kingoflolz/mesh-transformer-jax

Model parallel transformers in JAX and Haiku

拖动卡牌调整角度，点击翻面查看卡背。