anshul and i found a cool paper, so we reading group-ed it and did this implementation (without looking at any of their code!) it took 16 hours, spread over 2 days during winter break!
paper: https://arxiv.org/abs/2301.00808 notes from reading: https://docs.google.com/document/d/12B4mBNxzAIUF2buPERkJi-0YKxVLParzvB5PTIvVLqQ/edit?usp=sharing