This job is closed

We regret to inform you that the job you were interested in has been closed. Although this specific position is no longer available, we encourage you to continue exploring other opportunities on our job board.

Software Engineer, ML Compiler

$177,008 - $251,000/Yr

Meta - Redmond, WA

posted 2 months ago

Full-time - Mid Level
Redmond, WA
5,001-10,000 employees
Web Search Portals, Libraries, Archives, and Other Information Services

About the position

Meta Reality Labs is seeking a software engineer to support the development of a compiler tool-chain for advanced deep learning hardware components optimized for augmented and virtual reality systems. The role involves architecting, designing, and implementing a new compiler, contributing to a full stack development effort that compiles PyTorch models into binaries for hardware accelerators. The successful candidate will work alongside a team of experts in compilers, machine learning algorithms, and hardware design.

Responsibilities

  • Analyze and design effective compiler passes and optimizations.
  • Implement and/or enhance code generation targeting machine learning accelerators.
  • Work with algorithm research teams to map ML graphs to hardware implementations, model data-flows, create cost-benefit analysis and estimate silicon power and performance.
  • Contribute to the development of machine-learning libraries, intermediate representations, export formats, and analysis tools.
  • Conduct design and code reviews.
  • Evaluate code performance, debug, diagnose and drive resolution of compiler and cross-disciplinary system issues.
  • Analyze and improve the efficiency, scalability, and stability of our toolchains.
  • Interface with other compiler-focused teams to evaluate and incorporate their innovations and vice versa.
  • Mentor other engineers on improving the quality of engineering work in the broader team.

Requirements

  • Bachelor's degree in Computer Science, Computer Engineering, relevant technical field, or equivalent practical experience.
  • 3+ years experience writing compilers or similar code optimization software.
  • Experience in software design and programming experience in Python and/or C/C++ for development, debugging, testing and performance analysis.
  • Experience crossing multi-disciplinary boundaries to drive optimal system solutions.
  • Experience in AI framework development or accelerating deep learning models on hardware architectures.

Nice-to-haves

  • Experience in deep learning algorithms and techniques, e.g., convolutional neural networks, recurrent networks, etc.
  • Experience of developing in a mainstream machine-learning framework, e.g. PyTorch, Tensorflow or Caffe.
  • Experience working on and contributing to an active compiler toolchain codebase, such as LLVM, MLIR, GCC, MSVC, Glow.
  • Experience with machine-code generation or compiler back-ends.
  • Experience working and communicating cross functionally in a team environment.
  • Experience driving large efforts across multiple teams.

Benefits

  • Competitive salary and bonus structure
  • Equity options
  • Comprehensive health insurance
  • Retirement savings plan
  • Paid time off and holidays
  • Professional development opportunities
© 2024 Teal Labs, Inc
Privacy PolicyTerms of Service