JAXにできてPyTorchにできないこと
zenn.dev