Software Engineer, AI Systems

Job Summary

Innovating in AI systems technologies, the full-time Software Engineer, AI Systems will develop libraries and GPU kernel technologies to enhance AI inference, collaborating with engineers in a remote or onsite environment.

Key responsibilities:

• Developing new AI systems technologies for efficient inference
• Designing and optimizing kernels for high-impact AI workloads
• Building extensible abstractions for LLM serving engines and contributing to open source communities

Required qualifications:

• Master's degree in Computer Science, Electrical Engineering, or related field (PhD preferred)
• 2+ years of experience in ML/DL systems development
• Strong experience with deep learning frameworks (e.g., PyTorch, JAX, TensorFlow)
• Proficiency in Python and C/C++ programming
• Experience in GPU kernel development and performance optimizations using CUDA C/C++