Software Engineer, Systems ML

1 month ago

Boston, United States Meta Inc Full time

Summary:

In this role, you will be a member of the MTIA (Meta Training & Inference Accelerator) Software team and part of the bigger industry-leading PyTorch AI framework organization. MTIA Software Team has been developing a comprehensive AI Compiler strategy that delivers a highly flexible platform to train & serve new DL/ML model architectures, combined with auto-tuned high performance for production environments across specialized hardware architectures. The compiler stack, DL graph optimizations, and kernel authoring for specific hardware, directly impacts performance and deployment velocity of both AI training and inference platforms at Meta.You will be working on one of the core areas such as PyTorch framework components, AI compiler and runtime, high-performance kernels and tooling to accelerate machine learning workloads on the current & next generation of MTIA AI hardware platforms. You will work closely with AI researchers to analyze deep learning models and lower them efficiently on MTIA hardware. You will also partner with hardware design teams to develop compiler optimizations for high performance. You will apply software development best practices to design features, optimization, and performance tuning techniques. You will gain valuable experience in developing machine learning compiler frameworks and will help in driving next generation hardware software codesign for AI domain specific problems.

Required Skills:

Software Engineer, Systems ML - Frameworks / Compilers / Kernels Responsibilities:

  1. Development of SW stack with one of the following core focus areas: AI frameworks, compiler stack, high performance kernel development and acceleration onto next generation of hardware architectures.

  2. Contribute to the development of 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.

  3. Analyze deep learning networks, develop & implement compiler optimization algorithms.

  4. Collaborating with AI research scientists to accelerate the next generation of deep learning models such as Recommendation systems, Generative AI, Computer vision, NLP etc.

  5. Performance tuning and optimizations of deep learning framework & software components.

Minimum Qualifications:

Minimum Qualifications:

  1. Proven C/C++ programming skills

  2. Currently has, or is in the process of obtaining a Bachelor's degree in Computer Science, Computer Engineering, relevant technical field, or equivalent practical experience. Degree must be completed prior to joining Meta.

  3. Experience in AI framework development or accelerating deep learning models on hardware architectures.

Preferred Qualifications:

Preferred Qualifications:

  1. A Bachelor's degree in Computer Science, Computer Engineering, relevant technical field and 4+ years of experience in AI framework development or accelerating deep learning models on hardware architectures OR a Master's degree in Computer Science, Computer Engineering, relevant technical field and 2+ years of experience in AI framework development or accelerating deep learning models on hardware architectures OR a PhD in Computer Science Computer Engineering, or relevant technical field.

  2. Knowledge of GPU, CPU, or AI hardware accelerator architectures.

  3. Experience working with frameworks like PyTorch, Caffe2, TensorFlow, ONNX, TensorRT

  4. OR AI high performance kernels: Experience with CUDA programming, OpenMP / OpenCL programming or AI hardware accelerator kernel programming. Experience in accelerating libraries on AI hardware, similar to cuBLAS, cuDNN, CUTLASS, HIP, ROCm etc.

  5. OR AI Compiler: Experience with compiler optimizations such as loop optimizations, vectorization, parallelization, hardware specific optimizations such as SIMD. Experience with MLIR, LLVM, IREE, XLA, TVM, Halide is a plus.

  6. OR AI frameworks: Experience in developing training and inference framework components. Experience in system performance optimizations such as runtime analysis of latency, memory bandwidth, I/O access, compute utilization analysis and associated tooling development.

Public Compensation:

$56.25/hour to $173,000/year + bonus + equity + benefits

Industry: Internet

Equal Opportunity:

Meta is proud to be an Equal Employment Opportunity and Affirmative Action employer. We do not discriminate based upon race, religion, color, national origin, sex (including pregnancy, childbirth, or related medical conditions), sexual orientation, gender, gender identity, gender expression, transgender status, sexual stereotypes, age, status as a protected veteran, status as an individual with a disability, or other applicable legally protected characteristics. We also consider qualified applicants with criminal histories, consistent with applicable federal, state and local law. Meta participates in the E-Verify program in certain locations, as required by law. Please note that Meta may leverage artificial intelligence and machine learning technologies in connection with applications for employment.

Meta is committed to providing reasonable accommodations for candidates with disabilities in our recruiting process. If you need any assistance or accommodations due to a disability, please let us know at accommodations-ext@fb.com.

  • AI/ML Engineer

    11 hours ago

    Boston, United States Comtech LLC Full time

    AI/ML Engineer Atlanta, GA (On-site) Contract-2-Hire Job Purpose The Artificial Intelligence/Machine Learning (AI/ML) Engineer develops AI/ML algorithms, cloud computing, and/or heterogeneous distributed computing infrastructures to support the deployment of AI/ML applications. The AI/ML Engineer also researches the mathematical foundations and frameworks...

  • Software Engineer III

    6 days ago

    Boston, United States Chewy Full time

    Our Opportunity: Chewy is seeking a Software Engineer III (AI/ML) in one of our hub locations for our customer service technology team. This is a high-profile position that will have exposure across the entire business, influencing the vision and implementation of architecture, design and features for an AI platform. Chewy's mission is to empower pet parents...

  • AI/ML - Innovation Software Engineer

    6 days ago

    Boston, United States Della Infotech Full time

    Title: AI/ML - Innovation Software Engineer Location: Boston, MA 4 days/week onsite In person interview required All visa's are OK Job Description: As an Innovation Software Engineer, you will be at the forefront of driving technological advancements within our organization. You will be responsible for leveraging emerging technologies, such as GenAI and...

  • AI/ML - Innovation Software Engineer

    1 month ago

    Boston, United States Della Infotech Full time

    Title: AI/ML - Innovation Software Engineer Location: Boston, MA 4 days/week onsite In person interview required All visa's are OK Job Description: As an Innovation Software Engineer, you will be at the forefront of driving technological advancements within our organization. You will be responsible for leveraging emerging technologies, such as GenAI and...

  • AI/ML - Innovation Software Engineer

    6 days ago

    Boston, United States Della Infotech Full time

    Title: AI/ML - Innovation Software Engineer Location: Boston, MA 4 days/week onsite In person interview required All visa's are OK Job Description: As an Innovation Software Engineer, you will be at the forefront of driving technological advancements within our organization. You will be responsible for leveraging emerging technologies, such as GenAI and...

  • Director of Software Engineering

    1 month ago

    Boston, United States Discover International Full time

    Discover International is supporting an exciting AI Drug Discovery Company to grow its engineering team. Our client is looking to hire a Director of Software Engineering. We are seeking a seasoned Director of Software Engineering to lead our software engineering teams focused on AI-driven drug discovery. The ideal candidate will combine expertise in...

  • Senior Software Engineer

    4 weeks ago

    Boston, United States WHOOP Full time

    At WHOOP, we're on a mission to unlock human performance. WHOOP empowers members to perform at a higher level through a deeper understanding of their bodies and daily lives. We are looking for a highly skilled Senior Software Engineer to join our MLOps team, focusing on the development and optimization of ML cloud infrastructure. In this role, you will play...

  • Senior Software Engineer

    7 hours ago

    Boston, United States WHOOP Full time

    At WHOOP, we're on a mission to unlock human performance. WHOOP empowers members to perform at a higher level through a deeper understanding of their bodies and daily lives. We are looking for a highly skilled Senior Software Engineer to join our MLOps team, focusing on the development and optimization of ML cloud infrastructure. In this role, you will play...

  • Senior Software Engineer

    6 days ago

    Boston, United States WHOOP Full time

    At WHOOP, we're on a mission to unlock human performance. WHOOP empowers members to perform at a higher level through a deeper understanding of their bodies and daily lives. We are looking for a highly skilled Senior Software Engineer to join our MLOps team, focusing on the development and optimization of ML cloud infrastructure. In this role, you will play...

  • Senior Software Engineer

    1 month ago

    Boston, United States WHOOP Full time

    At WHOOP, we're on a mission to unlock human performance. WHOOP empowers members to perform at a higher level through a deeper understanding of their bodies and daily lives. We are looking for a highly skilled Senior Software Engineer to join our MLOps team, focusing on the development and optimization of ML cloud infrastructure. In this role, you will play...

  • Sr. Engineer: AI/ML

    2 months ago

    boston, United States BigRio Full time

    Sr. Engineer: AI/ML Location: Boston, MA- OnsiteDuration - Full TimeAbout BigRio:BigRio is a remote-based, technology consulting firm with headquarters in Boston, MA. We deliver solutions ranging from: custom development, software implementation, data analytics, and machine learning/AI integrations. We are a one-stop shop that attracts clients from a variety...

  • Sr. Engineer: AI/ML

    4 months ago

    Boston, United States BigRio Full time

    Sr. Engineer: AI/ML Location: Boston, MA- OnsiteDuration - Full TimeAbout BigRio:BigRio is a remote-based, technology consulting firm with headquarters in Boston, MA. We deliver solutions ranging from: custom development, software implementation, data analytics, and machine learning/AI integrations. We are a one-stop shop that attracts clients from a variety...

  • Sr. Engineer: AI/ML

    2 months ago

    boston, United States BigRio Full time

    Sr. Engineer: AI/ML Location: Boston, MA- OnsiteDuration - Full TimeAbout BigRio:BigRio is a remote-based, technology consulting firm with headquarters in Boston, MA. We deliver solutions ranging from: custom development, software implementation, data analytics, and machine learning/AI integrations. We are a one-stop shop that attracts clients from a variety...

  • Software Engineer, Machine Learning

    6 days ago

    South Boston, United States Meta Inc Full time

    Summary: Meta is embarking on the most transformative change to its business and technology in company history, and our Machine Learning Engineers are at the forefront of this evolution. By leading crucial projects and initiatives that have never been done before, you have an opportunity to help us advance the way people connect around the world. The ideal...

  • Principal Software Engineer

    3 weeks ago

    Boston, United States BigTime Software Full time

    BigTime Software is looking to add a dynamic Principal Software Engineer to our development team in support of our Projector PSA product line. The individual in this role will collaborate extensively to design, build, and test functionality in our award-winning, cloud based system. The ideal candidate will be passionate about development and able to solve...

  • Director of Software Engineering

    2 weeks ago

    Boston, United States ZipRecruiter Full time

    Job DescriptionCompany Overview1910 Genetics is a Series A stage biotechnology company that is pioneering a novel Input-Transform-Output (ITO) platform as the first ever horizontal AI infrastructure for drug discovery.In a crowded space with hundreds of AI drug discovery companies, 1910's ITO platform is differentiated by:Modality agnostic drug discovery:...

  • Software Engineering Manager, Bidding

    4 weeks ago

    Boston, Massachusetts, United States Wayfair Full time

    About the Role:We are seeking a highly skilled Software Engineering Manager to lead our Bidding and Decisioning team. As a key member of our engineering organization, you will be responsible for managing a high-performance engineering team, partnering with data science and product teams to define the direction for our platform, and owning the technical...

  • AI Software Engineer

    7 days ago

    Boston, United States Slalom Full time

    **This role is not eligible for 100% remote work. Employees must live within a commutable distance of the Boston or Hartford offices.** AI Software Engineer - Consultant and Senior Consultant We are seeking an experienced AI Software Engineer with a strong background in Python development, API building, and proficiency in AI/ML frameworks. The ideal...

  • AI Software Engineer

    1 month ago

    Boston, United States Slalom Full time

    **This role is not eligible for 100% remote work. Employees must live within a commutable distance of the Boston or Hartford offices.** AI Software Engineer - Consultant and Senior Consultant We are seeking an experienced AI Software Engineer with a strong background in Python development, API building, and proficiency in AI/ML frameworks. The ideal...

  • AI Software Engineer

    7 days ago

    Boston, United States Slalom Full time

    **This role is not eligible for 100% remote work. Employees must live within a commutable distance of the Boston or Hartford offices.** AI Software Engineer - Consultant and Senior Consultant We are seeking an experienced AI Software Engineer with a strong background in Python development, API building, and proficiency in AI/ML frameworks. The ideal...