AI Infrastructure Engineer
What you can expect
We are seeking an experienced AI Infrastructure Engineer to join our AI Incubation team. You will be focused on building and optimizing large-scale training infrastructure for Large Language Models (LLMs). The ideal candidate will combine engineering fundamentals with practical experience in AI infrastructure development, demonstrating both technical depth and the ability to deliver scalable solutions for complex AI systems.
About the Team
The AI incubation team is dedicated to incubating AI breakthroughs, including foundational AI techniques and AI native applications that will largely improve people's work productivity.
Responsibilities:
- Designing and developing scalable AI infrastructure solutions for training and deploying large language models
- Building and optimizing distributed training platforms using cutting-edge technologies
- Implementing and maintaining containerized AI environments using Docker and Kubernetes
- Optimizing CUDA kernels for maximum GPU utilization and performance
- Developing platform software to support AI/ML workflows
- Collaborating with AI researchers to implement efficient training and inference pipelines
What we're looking for
- Have a bachelor's degree in Computer Science, Engineering, AI, Machine Learning, Distributed System or related field
- 5+ years of software engineering experience with focus on infrastructure and systems
- Have expertise in GPU programming and CUDA optimization
- Have experience with container technologies (Docker, Kubernetes), distributed systems and cloud computing
- Demonstrate experience building large-scale distributed systems and optimizing neural network performance
- Possess programming skills in Python, C++, and CUDA, with deep learning frameworks (PyTorch, Transformers)
Salary Range or On Target Earnings
Not included in the source posting: about the role, what you'll do, benefits.
Skills
Who can apply
The employer didn't state any visa, work authorization, citizenship or clearance requirements in this posting. Confirm with the employer before applying.
Read automatically from the employer's posting text. Always confirm with the employer — requirements can change after a job is published.