https://kidger.site/thoughts/torch2jax/ Learning JAX as a PyTorch developerPatrick KidgerPosted on November 9, 2023 “Assuming you already know PyTorch, this is what I’ve found that you need to know to get up
What I Read: Engineering Trade-Offs in Automatic Differentiation
https://www.stochasticlifestyle.com/engineering-trade-offs-in-automatic-differentiation-from-tensorflow-and-pytorch-to-jax-and-julia/ Engineering Trade-Offs in Automatic Differentiation: from TensorFlow and PyTorch to Jax and JuliaDecember 25 2021Author: Christopher Rackauckas “To understand the differences between automatic differentiation libraries, let’s talk about the
What I Read: Tonks Multi-Task Model
https://medium.com/shoprunner/tonks-building-one-multi-task-model-to-rule-them-all-3e5d020f1f2b Tonks: Building One (Multi-Task) Model to Rule Them All!Co-written by Nicole Carlson and Michael SugimuraMichael SugimuraApr 28 “Tonks is a library that streamlines the training of multi-task PyTorch networks.