Qualifications

• Proficiency in ML model development and optimization techniques on GPUs or AI accelerators
• Strong understanding of deep learning algorithms and software engineering
• Experience with neural networks in TensorFlow or PyTorch
• Ability to thrive in a fast-paced work environment
• Excellent communication skills for conveying complex technical concepts
• Attention to detail and introspective mindset
• Proficiency in Python

Responsibilities

• Research, prototype, and evaluate state-of-the-art model optimization techniques
• Characterize neural network performance using data and profiling
• Incorporate best practices into the ML development lifecycle
• Define the technical vision for deep learning model optimizations
• Write technical reports to communicate results to colleagues and customers
• Develop, deploy, and optimize deep learning models on various platforms

Benefits

• Competitive health insurance options
• 401K plan management
• Free lunch and fully-stocked kitchen at the South Bay office
• Monthly wellness stipend and office setup allowance
• Company retreats and additional perks as the company scales
• Opportunity to work on impactful problems in the field