2027 Summer Intern MS PhD Software Engineer ML Frameworks and Efficiency
Posted Updated
Description
You will support machine learning frameworks and efficiency initiatives for LPM model training. You will understand the current model’s problem space, extend reinforcement learning infrastructure, profile and address training bottlenecks, and evaluate and integrate promising reinforcement learning technologies.
Responsibilities
- Understand the current LPM model and its requirements
- Improve and extend reinforcement learning infrastructure to support LPM model training
- Profile training performance and improve training efficiency
- Identify and integrate promising reinforcement learning technologies
Requirements
- Currently pursuing a Master's or PhD degree in Computer Science, Electrical Engineering, Machine Learning, Robotics, or a related technical field
- Fundamental understanding of machine learning, deep learning, and reinforcement learning
- Familiarity with JAX, Flax, PyTorch, or TensorFlow
- Strong Python software development skills
Benefits
- Housing or relocation bonus, if applicable
- Medical, dental, and vision insurance
- Intern events and networking opportunities
- Free breakfast, lunch, dinner, and snacks
- Free access to Google shuttles
- Onsite gym