GitHub repository: martinResearch/onnxscript
Description: ONNX Script enables developers to naturally author ONNX functions and models using a subset of Python.