This should match setup.py Co-authored-by: hauntsaninja <> Co-authored-by: Philippe Tillet <phil@openai.com>