about 6 hours ago
Taipei, Taiwan
Mid Level / Senior
H1B Sponsor
Responsibilities
- Design and implement technical solutions for performance optimization of individual ML models.
- Execute end-to-end tasks consistently and in a timely manner, contributing to larger performance goals for ML accelerators.
- Collect, trace, and analyze application/ML model performance data to identify optimization opportunities.
- Collaborate with infrastructure teams to implement performance-critical optimizations.
- Contribute to design documents, participate in code reviews, and write comprehensive unit tests.
- Proactively suggest areas for future performance work within the team's scope.
Requirements
- BS degree in Computer Science/Electrical Engineering or equivalent practical experience.
- 3+ years of experience writing complex C++ code.
- 3+ years of experience writing code in Python.
- 1+ years of experience in optimizing compute performance for ML applications.
- Experience in compute architectures and performance analysis optimization methodologies.
Benefits
- Eligible for Waymo’s discretionary annual bonus program.
- Participation in equity incentive plan.
- Generous Company benefits program.
Tech Stack
C++Python
Categories
AI & MLData Engineering