AI Co-Design Engineer - ML Acceleration for AR/VR

Meta; Sunnyvale (CA); On-site; USD 184,000 - 257,000; Full time

Job summary

Meta’s Reality Labs seeks a Software Engineer to advance AI co-design and ML acceleration for AR/VR hardware. You will co-design algorithms, hardware and software stacks to enable training and on-device inference for vision and language models, across edge devices.

The role emphasizes collaboration with researchers, system-level optimization, and contributing to patents and publications in leading venues while aligning with Meta’s AR/VR roadmap.

Qualifications

• Bachelor's degree in Computer Science, Computer Engineering, or equivalent practical experience.
• PhD in Electrical Engineering, Computer Science, or equivalent experience.
• Experience developing AI-System infrastructure, AI algorithms or AI hardware acceleration in C/C++ or Python.

Responsibilities

• Identify and solve multi-discipline ML acceleration problems involving algorithms, network design, hardware architecture and AR/VR use cases.
• Work across hardware and software to solve co-design problems with researchers.
• Codesign and invent novel ML accelerator and system architecture solutions.
• Develop state-of-the-art model compression and scalability techniques using Numerics, pruning, distillation, etc.
• Optimize models on hardware accelerators to meet latency and power constraints.
• Influence partners to adopt solutions through data-driven trade-off analysis.
• Define use cases and develop evaluation benchmarks for approaches.
• Apply ML acceleration knowledge to other system components.
• Attend conferences and stay updated with ML acceleration advancements.
• Contribute to patents and publications.

Skills

AI systems infra

C/C++

Python

ML acceleration

Education

Bachelor's degree in Computer Science/Engineering

PhD in Electrical Engineering or Computer Science

Tools

PyTorch

TensorFlow

C/C++

Python

Job description

Meta’s Reality Labs seeks a Software Engineer to advance AI co-design and ML acceleration for AR/VR hardware. You will co-design algorithms, hardware and software stacks to enable training and on-device inference for vision and language models, across edge devices.

The role emphasizes collaboration with researchers, system-level optimization, and contributing to patents and publications in leading venues while aligning with Meta’s AR/VR roadmap.