tech⚡PyTorch to JAX 移行ガイド（GANの学習｜TrainStateのカスタマイズ編）jaxPyTorchdeeplearningPythonjellied_unagi3ヶ月前23 min read 6
tech⚡PyTorch to JAX 移行ガイド（GPUでのCNN学習 | BatchNorm編）deeplearningjaxPyTorchPythonjellied_unagi4ヶ月前15 min read 16
