Dec 26, 2024 Released writeup and code for a fast RG-LRU kernel Mar 31, 2023 Released post detailing my implementation of ZeRO optimizer sharding in Jax