add compute capability from python frontend to backend. Co-authored-by: Keren Zhou <kerenzhou@openai.com>