2027 Summer Intern MS PhD Machine Learning Engineer
Posted Updated
Description
You will train and fine-tune a multi-task transformer using JAX and Flax on TPUs, design offline and end-to-end evaluations, build data and evaluation pipelines, deliver production-quality code, and communicate results through technical documentation and presentations.
Responsibilities
- Train and fine-tune a multi-task transformer over driving-log sequences
- Design and run offline and end-to-end evaluations
- Build datasets and evaluation pipelines
- Deliver production-quality code in a shared codebase
- Communicate results through a design document, deep dives, and a final presentation
Requirements
- Enrollment in a PhD or MS program in Computer Science, Machine Learning, or a related field
- Deep learning model training and evaluation
- JAX, PyTorch, or TensorFlow
- Data pipeline development
- C++
- Python
- Machine learning fundamentals
- Experiment design
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
- Mental wellness support
- Gym membership
- Wellness programs
- Bonus opportunities
- Equity
- Employee discounts
- Paid time off
- Bereavement leave
- Sick leave
- Parental leave
- Education reimbursement
- Employee Resource Groups
- Volunteer time off