I'm Ashish Rao, a software engineer at Nvidia working on JAX (a Python library to write hardware-accelerated numerical programs). I'm broadly interested in AI, programming languages, compilers, and distributed systems. Check out my blog, and feel free to say hello via email to ashish dot arartc at gmail.com.