有沒有人在 Jax 和 Pytorch 中進行大規模訓練的經驗?如果有的話,在什麼規模下你真的感受到 Pytorch 的痛點,而在什麼規模下你覺得 Jax 的一般痛點絕對是值得的?
21.42K