Meta Platforms, Inc. (Meta), formerly known as Facebook Inc., builds technologies that help people connect, find communities, and grow businesses. When Facebook launched in 2004, it changed the way people connect. Apps and services like Messenger, Instagram, and WhatsApp further empowered billions around the world. Now, Meta is moving beyond 2D screens toward immersive experiences like augmented and virtual reality to help build the next evolution in social technology. To apply, click “Apply to Job” online on this web page.

Software Engineer, Machine Learning Responsibilities

• Research, design, develop and test comprehensive AI Compiler strategy that delivers a highly flexible platform to train & serve massive social data and prediction DL/ML models.
• Contribute to the industry-leading PyTorch AI framework core compilers to support new state of the art inference and training AI hardware accelerators and optimize their performance.
• Have industry experience working on a range of ranking, classification, recommendation, and optimization problems, e.g. click-through or conversion rate prediction, ads/feed/search ranking, text/sentiment classification or collaborative filtering/recommendation.
• Suggest, collect, analyze and synthesize bottlenecks within DL/ML models.
• Identify optimization candidates in DL/ML model graph to accelerate the inference and training process.
• Adapt standard machine learning methods to best exploit modern parallel environments, including distributed clusters, GPUs and custom AI accelerators.
• Develop, generalize and optimize the E2E compiler stack for Meta’s custom domain-specific language (DSL) for AI accelerators.
• Author high- performance kernels with Meta’s custom domain-specific language to support more PyTorch operators used in the next generation of DL models.
• Analyze the graph representation of deep learning models to identify possible optimization candidates.
• Develop and apply kernel fusion techniques and memory optimization algorithms accordingly.
• Collaborate with engineers from the hardware team and AI research scientists to onboard the next generation of DL/ML models, such as Recommendation systems, Generative AI and Computer vision models, onto Meta’s custom AI accelerators.

Minimum Qualifications

• Requires a Master’s degree in Computer Science, Computer Software, Computer Engineering, Applied Sciences, Mathematics, Physics, or related field. Requires completion of a university-level course, research project, internship, or thesis in the following:
• Machine Learning Framework(s): PyTorch, MXNet, or Tensorflow
• Knowledge of GPU, CPU, or AI accelerator hardware architectures
• Experience with the design and implementation of compilers, including parsing, semantic analysis, compiler optimization and code generation.
• Experience with high-performance programming, including CUDA programming, OpenMP/OpenCL programming
• Knowledge of Machine learning, recommendation systems, computer vision or distributed systems
• Translating insights into business recommendations
• Developing and debugging in C, C++ and Java
• Scripting languages: Perl, Python, PHP, or shell scripts
• Relational databases and SQL
• Software development tools: Code editors (VIM, Emacs or VS Code), and revision control systems (Subversion, GIT, Mercurial or Perforce)
• Linux, UNIX, or other *nix-like OS as evidenced by file manipulation, advanced commands, and shell scripting
• Build highly-scalable performant solutions
• Data processing, programming languages, operating systems
• Applying algorithms and core computer science concepts to real world systems as evidenced by recognizing and matching patterns from different areas of computer science in production systems
• Distributed systems.

About Meta

Meta builds technologies that help people connect, find communities, and grow businesses. When Facebook launched in 2004, it changed the way people connect. Apps like Messenger, Instagram and WhatsApp further empowered billions around the world. Now, Meta is moving beyond 2D screens toward immersive experiences like augmented and virtual reality to help build the next evolution in social technology. People who choose to build their careers by building with us at Meta help shape a future that will take us beyond what digital connection makes possible today—beyond the constraints of screens, the limits of distance, and even the rules of physics. Meta is committed to providing reasonable support (called accommodations) in our recruiting processes for candidates with disabilities, long term conditions, mental health conditions or sincerely held religious beliefs, or who are neurodivergent or require pregnancy-related support. If you need support, please reach out to accommodations-ext@fb.com. US$169,442/year to US$200,200/year + bonus + equity + benefits

Individual compensation is determined by skills, qualifications, experience, and location. Compensation details listed in this posting reflect the base hourly rate, monthly rate, or annual salary only, and do not include bonus, equity or sales incentives, if applicable. In addition to base compensation, Meta offers benefits. Learn more about benefits at Meta.