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....

September 15, 2020 · 4 min · 683 words · Me