Giters
fattorib
/
transformer_shmap
Tensor Parallelism with JAX + Shard Map
Geek Repo:
Geek Repo
Github PK Tool:
Github PK Tool
Stargazers:
10
Watchers:
1
Issues:
0
Forks:
1
fattorib/transformer_shmap Stargazers
Abel Legese
Abellegese
Amr Kayid
AmrMKayid
Mahmoud Asem
ASEM000
Paul Balanca
balancap
Derrick
flrngel
Roy Frostig
froystig
Jon Deaton
jondeaton
Matthew Johnson
mattjj
Mit Kotak
mitkotak
snoop2head
snoop2head
