Update JAX AI for PyTorch users tutorial - #135

Update JAX AI for PyTorch users tutorial#135

8bitmp3 wants to merge 4 commits into

jax-ml:mainjax-ml/jax-ai-stack:mainfrom

8bitmp3:update-jax-for-pytorch8bitmp3/jax-ai-stack:update-jax-for-pytorchCopy head branch name to clipboard