2027 PhD AI Model Optimization & Software Engineer Intern/Co-op
Core
Develop, benchmark, and optimize AI software for training, fine-tuning, and inference across CPU, GPU, and accelerator platforms.
Role type
PhD AI Model Optimization & Software Engineer Intern
Builds
AI software, frameworks, libraries, execution runtimes, and deployment technologies
Domain
AI/ML, GPU computing, distributed systems, compiler optimization
Deliverable
production ML models
Required skills
Python, C/C++, AI frameworks (PyTorch, TensorFlow, JAX, ONNX Runtime, vLLM, SGLang), GPU kernel development (HIP, CUDA, OpenCL, Triton), model optimization (quantization, pruning, distillation), distributed computing, MLOps/DevOps
Preferred skills
Research publications, preprints, open-source contributions, AI-assisted coding tools
Technologies
HIP, CUDA, OpenCL, Triton, PyTorch, TensorFlow, JAX, ONNX Runtime, vLLM, SGLang, Git, CMake, Docker
Responsibilities
Profile AI workloads and implement performance improvements; Design and optimize GPU/CPU kernels; Research and implement model optimization methods; Contribute to AI frameworks and runtimes; Develop parallel and distributed computing methods; Build automated evaluation systems and CI/CD pipelines
Seniority
PhD student, Intern
