You will:- Research, implement, and evaluate state-of-the-art generative models for planning and predictions using JAX
- Conduct machine learning models experiments
- Produce high-quality, thoroughly tested code
- Collaborate with research, product, and engineering teams to introduce new research concepts
You have:- Currently pursuing a PhD degree in Computer Science, Machine Learning, Robotics, or related field
- Experience with deep learning, or reinforcement learning
- Hands-on experience with modern deep learning libraries (eg: TensorFlow, JAX, Pytorch) with basic programming skills (eg: Python, C/C++)
- Strong analytical and debugging skills
We prefer:- Publications in top-tier conferences such as: ICML, NeurIPS, CVPR, ICCV, ECCV, ICLR, IROS, CoRL, ACL, or EMNLP
- Prior software development or research industry experience (including internships)
General Perks- Competitive compensation packages with a housing/relocation bonus (if applicable)
- Medical, dental, and vision insurance
- Fun intern events and networking opportunities
Onsite Perks- Free breakfast, lunch, dinner, and snacks
- Free access to Google shuttles
- Onsite gym
Note: This will be a hybrid onsite internship position. We will accept resumes on a rolling basis until the role is filled. To be in consideration for multiple roles, you will need to apply to each one individually - please apply to the top 3 roles you are interested in.
The expected hourly rate for this full-time position is listed below. Interns are also eligible to participate in the Company's generous benefits programs, subject to eligibility requirements.
Hourly PhD Pay
$85-$85 USD
