github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-10-16 04:11:24 +00:00
Code Issues Packages Projects Releases Wiki Activity
545 Commits 59 Branches 51 Tags
a0e59716920b4225fe0c226a9618c46f8ff25f0d
Commit Graph

4 Commits

Author SHA1 Message Date
Jiarui Fang
1190b2c4a4 [tensor] add cross_entrophy_loss (#868) 2022-04-25 16:01:52 +08:00
Frank Lee
1258af71cc [ci] cache cuda extension (#860) 2022-04-25 10:03:47 +08:00
Ziyue Jiang
bcc8655021 [Tensor ] Add 1Drow weight reshard by spec (#854) 2022-04-24 18:30:20 +08:00
Jiarui Fang
62f059251b [Tensor] init a tp network training unittest (#849) 2022-04-24 16:43:44 +08:00