Deep Dive: PyTorch vmap vs JAX vmap - Part 1
Deep Dive part 1: PyTorch vmap vs JAX vmap Introduction: Hello everyone, I am starting a new series of blog posts called deep dives, where I will do a deep and detailed analysis of machine learning frameworks and algorithms. Here I will break down new APIs and features of frameworks such as PyTorch, JAX, and tinygrad. In addition to that, I will also write about new algorithms in RLHF and other areas of research....