Meta

Software Engineer, AI Co-design

Meta
Apply
2 days ago
Sunnyvale, CA, USASenior
H1B sponsor

Base Salary

$184k - $257k/yr

Responsibilities

  • Identify and solve multidisciplinary ML acceleration problems involving algorithms, network design, hardware architecture, and AR/VR use cases.
  • Co-design ML accelerator and system architecture solutions and integrate algorithms and software to use those enhancements.
  • Develop model compression and scalability techniques including numerics, pruning, and distillation.
  • Optimize models on hardware accelerators under real-time latency and power constraints.
  • Define use cases, methodologies, and benchmarks for evaluating alternative approaches.
  • Analyze performance, power, and latency trade-offs and influence partners to adopt recommended solutions.
  • Stay current with ML acceleration research and contribute to patents and peer-reviewed publications.

Requirements

  • Bachelor's degree in Computer Science, Computer Engineering, a relevant technical field, 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.
  • Experience or knowledge of on-device algorithms, hardware-aware ML models, or ML compiler optimization for AI accelerators is preferred.
  • Experience or knowledge of training or inference for large-scale computer vision or language models is preferred.
  • Experience with PyTorch, TensorFlow, or similar machine learning toolsets is preferred.
  • Research and engineering experience through internships, work experience, coding competitions, or open-source contributions is preferred.
  • Experience evaluating system or algorithm designs using performance, power, and latency trade-offs is preferred.
  • A record of significant results through grants, fellowships, patents, or publications at leading workshops, journals, or conferences is preferred.
  • Experience with responsible and ethical AI practices and ongoing AI skill development is preferred.

Tech Stack

Categories

Meta

About Meta

10,000+ employees

Meta builds social platforms and communication apps—including Facebook, Instagram, WhatsApp, and Messenger—and develops AR/VR hardware and software such as Quest to power immersive computing. It monetizes primarily through advertising tools for businesses, with additional revenue from devices and services, and operates a massive global infrastructure. Founded in 2004 and headquartered in Menlo Park, California, Meta Platforms, Inc. is a public company traded on Nasdaq under the ticker META.

Contact me