If possible, could you please implement this? https://github.com/hao-ai-lab/JetSpec
If possible, could you please implement this? https://github.com/hao-ai-lab/JetSpec