PyTorch AI Acceleration Team Lead

3 weeks ago


Menlo Park, California, United States META Full time
Job Summary
The Meta AI Infra team is seeking a skilled Software Engineering Manager to lead the development of the PyTorch MTIA (Meta Training & Inference Accelerator) software stack. As part of this role, you will be responsible for managing a team of domain experts in developing high-performance AI acceleration software for inference and training platforms.

About the Role
This is an exciting opportunity to work with a talented team of engineers on cutting-edge AI technologies. You will be responsible for operating strategically and tactically, developing the vision and strategy to help set direction for the team, while managing day-to-day software development. Effective communication and collaboration with peer engineering teams are essential in this role.

Responsibilities
- Manage a team of developers, helping them develop their careers, and assigning projects tailored to their skill levels, long-term skill development, personalities, and work styles
- Work closely with dedicated recruiting staff to expand the team, including sourcing candidates, interviewing candidates, and on-boarding new employees
- Develop AI inference and/or training software stack experience is required, with a strong background in deep learning frameworks such as PyTorch, TensorFlow, JAX, ONNX, MXNet, TensorRT etc

Requirements
- At least 2 years of experience in managing a software team in a fast-paced capacity
- Demonstrated experience recruiting, building, structuring, leading technical organizations, including performance management
- Experience working with AI hardware such as GPUs, deep learning accelerators is a plus

Benefits
As a member of the Meta team, you will receive a competitive salary of $177,000/year to $251,000/year, bonus, equity, and benefits. Meta is proud to be an Equal Employment Opportunity and Affirmative Action employer.

  • Menlo Park, California, United States Meta Inc Full time

    Technical DetailsThis role will focus on developing and optimizing AI compilers and accelerators for the PyTorch AI framework. The ideal candidate will have expertise in machine learning systems and a strong background in computer science.Key Skills- Experience with AI compilers and accelerators.- Strong programming skills in Python and C/C++.- Knowledge of...


  • Menlo Park, California, United States META Full time

    Job Summary">META is seeking a highly skilled Partner Engineer, Generative AI to join its team. As a key member of the AI Partner Engineering team, you will work closely with strategic partners, machine learning leaders, and cloud service providers to build and launch new Generative AI product services and experiences.">About the Role">The successful...


  • Menlo Park, California, United States META Full time

    Job Summary:We are seeking a highly skilled Software Engineer, Systems ML to join our team at Meta and play a key role in developing the next generation of PyTorch Compiler technology. As a member of our team, you will have the opportunity to work on cutting-edge projects, collaborate with leading experts in the field, and contribute to the growth and...


  • Menlo Park, California, United States Meta Inc Full time

    Job OverviewWe are seeking an experienced system engineer to join our team and contribute to the development of AI compilers and accelerators. This role will involve working with our researchers to advance the state-of-the-art in ML Compiler space.Key Responsibilities- Collaborate with researchers to develop new AI compiler technologies.- Design and...


  • Menlo Park, California, United States Robinhood Full time

    Democratizing Finance through AI-Powered InnovationsRoznovat's commitment to making finance accessible to all necessitates cutting-edge solutions. As a Senior Software Engineer on our Agentic AI team, you will spearhead the design and development of large-scale systems, leveraging LLMs to drive innovation. With a strong background in software engineering,...


  • Menlo Park, California, United States META Full time

    Job Title: ML Compiler SpecialistThe PyTorch Compiler team at Meta is dedicated to making PyTorch run faster and more resource-efficient. We are the driving force behind PT2, a major step forward in PyTorch's history. Our technologies have gained recognition since their release.We are committed to building a compiler that withstands the test of time while...


  • Menlo Park, California, United States Altera AI Full time

    Job Title: Research and Development EngineerAltera AI seeks a talented Research and Development Engineer to drive the advancement of AI research.You will play a pivotal role in bringing our innovative research to life by implementing and experimenting with the latest research techniques in AI and machine learning.Your responsibilities will include...


  • Menlo Park, California, United States Trust A-tech Full time

    We're seeking a highly skilled Senior Machine Learning Scientist to join our Generative AI team at Meta. This role involves developing cutting-edge generative AI algorithms and corresponding systems, analyzing and improving safety and robustness, conducting research, collaborating with teams, and publishing results.RequirementsPh.D. in Computer Science,...


  • Menlo Park, California, United States Altera AI Full time

    At Altera AI, we're pushing the boundaries of innovation and seeking a talented Cutting-Edge AI Researcher to join our team. This is a full-time position based on-site in Menlo Park, with a competitive salary and additional benefits.About UsWe're a cutting-edge research organization dedicated to bringing our groundbreaking ideas to life. Our Research...


  • Menlo Park, California, United States Genesys Cloud Services, Inc. Full time

    At Genesys Cloud Services, Inc., we empower organizations to drive business outcomes through exceptional customer experiences. Our AI-powered Experience Orchestration platform enables companies of all sizes to accelerate growth by delivering personalized interactions at scale.We are a global organization with over 6,000 employees who collaborate to achieve...

  • AI Optimization Lead

    2 weeks ago


    Menlo Park, California, United States Character Full time

    About the RoleWe are seeking an experienced Staff Data Scientist with deep expertise in experimentation and ML frameworks to join our team. Domain knowledge with large language models (LLMs) and post-training optimization is crucial for expanding our AI capabilities.Key Responsibilities:Design and implement search retrieval and ranking algorithms that...


  • Menlo Park, California, United States META Full time

    Job DescriptionIn this highly influential role, you will be responsible for driving execution of developing models and systems that are crucial to the end user experience.About the TeamYou will play a key role in building the core AI tech powering experiences on META's Smart Glasses, AR Glasses, and Quest devices.ResponsibilitiesDevelop technical vision,...


  • Menlo Park, California, United States Genesys Cloud Services, Inc. Full time

    Empowering Organizations through Innovative AI ExperiencesAbout the Role:At Genesys Cloud Services, Inc., we are dedicated to revolutionizing customer experiences through cutting-edge AI solutions. As an AI Solutions Architect Leader, you will play a pivotal role in shaping our strategy and vision for delivering personalized, empathetic, and scalable...


  • Menlo Park, California, United States META Full time

    Meta is a leading technology company seeking a highly skilled Technical Lead to join our Large Language Model (LLM) team. This role offers a unique opportunity to work on cutting-edge AI projects and contribute to the development of state-of-the-art LLMs.About the JobThis position requires a strong background in generative AI and NLP, with experience in...


  • Menlo Park, California, United States Diffuse Bio Full time

    About Diffuse Bio">We're a cutting-edge biotech company pushing the boundaries of AI in molecular design.">Salary Range: $140,000 - $170,000 per year">Job Description">Develop and extend our deep generative modeling toolkit to improve downstream applications in molecular design.Conduct deep learning experiments to enhance model performance and develop new...


  • Menlo Park, California, United States Pathway Full time

    About UsPathway is an innovative company that enables organizations to run contextualized ML models connected to ever-changing enterprise data.This role is part of our R&D team focused on attention-based models and advancing the state-of-the-art in machine learning research.We have a strong presence in mobility, IoT data, logs and transactions, and...


  • Menlo Park, California, United States Meta Inc Full time

    **About Meta Inc.**We are committed to advancing the field of artificial intelligence by making fundamental advances in technologies to help interact with and understand our world.As a Research Scientist Intern, AI Core Machine Learning, you will have an opportunity to make core algorithmic advances and apply your ideas at an unprecedented scale.**Job...

  • Engineering Manager

    4 weeks ago


    Menlo Park, California, United States Brio Digital Full time

    We're seeking a talented Engineering Manager - AI to lead our team in developing innovative AI solutions. As a key member of our engineering team, you'll work closely with the Founding Engineer and CTO to shape our technical direction.Your responsibilities will include leading cross-functional teams, mentoring junior engineers, and driving technical...


  • Menlo Park, California, United States META Full time

    Company Overview">META is a leading technology company that aims to bring people closer together. Our Host Networking team is responsible for building high-performance network solutions for our AI workloads, and we are seeking a highly technical manager to lead our AI Transport group.">Salary">The estimated salary for this role is $177,000/year to...


  • Menlo Park, California, United States Robinhood Full time

    **Company Overview**Robby hood Markets is a leading fintech company that's changing the way people interact with their finances. Our goal is to make financial markets more accessible and transparent for everyone.We're seeking an experienced AI Infrastructure Specialist to join our team. As an AI Infrastructure Specialist, you will design and build scalable...