GitHub repository: wanchaol/spmd
Primary language: Python
Description: spmd layer implementation