Research Scientist Intern, PyTorch Compiler Job at Meta (Menlo Park)

Research Scientist Intern, PyTorch Framework Performance

Our team’s mission is to make PyTorch models high-performing, deterministic and ...

Location

United States , Bellevue

Salary:

7650.00 - 12134.00 USD / Month

Meta

Expiration Date

Until further notice

Requirements

Currently has, or is in the process of obtaining, a PhD degree in the field of Computer Science or a related STEM field
Deep knowledge of transformer architectures, including attention, feed-forward layers, and Mixture-of-Experts (MoE) models
Strong background in ML systems research, with domain knowledge in MoE efficiency, such as routing, expert parallelism, communication overheads, and kernel-level optimizations
Hands-on experience writing GPU kernels using CUDA and/or cuteDSL
Working knowledge of quantization techniques and their impact on performance and accuracy
Must obtain work authorization in the country of employment at the time of hire, and maintain ongoing work authorization during employment

Job Responsibility

Design and evaluate communication-aware, kernel-aware, and quantization-aware MoE execution strategies, combining ideas such as expert placement, routing, batching, scheduling, and precision selection
Develop and optimize GPU kernels and runtime components for MoE workloads, including fused kernels, grouped GEMMs, memory-efficient forward and backward passes
Explore quantization techniques (e.g., MXFP8, FP8) in the context of MoE, balancing accuracy, performance, and hardware efficiency
Build performance models and benchmarks to analyze compute, memory, communication, and quantization overheads across different sparsity regimes
Run experiments on single-node and multi-node GPU systems
Collaborate with the open-source community to gather feedback and iterate on the project
Contribute to PyTorch (Core, Compile, Distributed) within the scope of the project
Improve PyTorch performance in general

Research Scientist Intern, PyTorch On-Device

PyTorch is Meta’s deep learning framework for fast, flexible AI/ML experimentati...

Location

United States , Bellevue

Salary:

7650.00 - 12134.00 USD / Month

Meta

Expiration Date

Until further notice

Requirements

Currently has, or is in the process of obtaining, PhD degree in the field of Computer Science or a related STEM field
Experience in ML compilers, sparsity, quantization, kernel development, or similar as applied to on-device and highly-constrained environments
Must obtain work authorization in the country of employment at the time of hire, and maintain ongoing work authorization during employment

Job Responsibility

Develop new or apply existing performance techniques to on-device AI
Explore quantization, sparsity, and model/software co-design as solutions
Apply knowledge and research to advance the state-of-the-art in on-device machine learning frameworks
Collaborate with users and developers of PyTorch and ExecuTorch to enable new use cases inside and outside Meta

Research Scientist Intern, AI & Compute Foundation - MTIA Software

The MTIA (Meta Training & Inference Accelerator) Software team is part of the AI...

Location

Canada , Toronto

Salary:

6240.00 - 10334.00 CAD / Month

Meta

Expiration Date

Until further notice

Requirements

Currently has, or is in the process of obtaining, PhD degree in the field of Computer Science or a related STEM field
C/C++ programming skills
Must obtain work authorization in country of employment at the time of hire, and maintain ongoing work authorization during employment
Knowledge of Computer Architecture and Distributed systems with interest in one or more of High Performance Computing, Numerics, Performance and AI hardware including compute, networking and storage

Job Responsibility

Development of Software 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
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
Analyze deep learning networks, develop & implement compiler optimization algorithms
Collaborating with AI research scientists to accelerate the next generation of deep learning models such as Recommendation systems, Generative AI, Computer vision, NLP etc
Performance tuning and optimizations of deep learning framework & software components

Research Scientist Intern, PyTorch Distributed

Meta is seeking a Research Scientist Intern to join our Meta PyTorch Distributed...

Location

United States , Menlo Park

Salary:

7650.00 - 12134.00 USD / Month

Meta

Expiration Date

Until further notice

Requirements

Currently has, or is in the process of obtaining, PhD degree in the field of Computer Science or a related STEM field
Experience in one or more of the following machine learning/deep learning domains: Large scale training and inference ML Systems Research, ML theory: Basic knowledge about ML models in different modalities like LLM (Large Language Models), Vision (VITS, MVITS) and Multimodal and how scale impacts performance, ML systems: AI infrastructure, machine learning accelerators, high performance computing, machine learning compilers, GPU architecture, machine learning frameworks, distributed systems, on-device optimization
Must obtain work authorization in the country of employment at the time of hire, and maintain ongoing work authorization during employment

Job Responsibility

Apply relevant AI and machine learning techniques to advance the state-of-the-art in machine learning frameworks
Collaborate with users of PyTorch to enable new use cases for the framework both inside and outside Meta
Develop novel, accurate AI algorithms and advanced systems for large scale distributed training and inference
Leverage graph-based and compiler-based technologies to optimize distributed training and distributed inference use-cases

Research Scientist Intern, MSL Infra Kernels & Optimizations

Meta’s Meta SuperIntelligence Labs (MSL) Infra Kernels & Optimizations (K&O) tea...

Location

United Kingdom , London

Salary:

Not provided

Meta

Expiration Date

Until further notice

Requirements

Currently has, or is in the process of obtaining, a PhD degree in the field of Computer Science, Computer Vision, Generative AI, NLP, relevant technical field, or equivalent practical experience
Must obtain work authorization in country of employment at the time of hire, and maintain ongoing work authorization during employment
Specialized experience in one or more of the following areas: Accelerators/GPU architectures, High Performance Computing (HPC), Machine Learning Compilers, Training/Inference ML Systems, Model Compression, Communication Collectives, ML Kernels/Operator optimizations, Machine learning frameworks (e.g. PyTorch) and SW/HW co-design
Experience developing AI-System infrastructure or AI algorithms in C/C++ or Python

Job Responsibility

Explore, prototype and productionize highly optimized ML kernels to unlock full potential of current and future accelerators for Meta’s AI workloads. Open source SOTA implementations as applicable
Explore, co-design and optimize parallelisms, compute efficiency, distributed training/inference paradigms and algorithms to improve the scalability, efficiency and reliability of inference and large-scale training systems
Optimize inference and training communications performance at scale and investigate improvements to algorithms, tooling, and interfaces, working across multiple accelerator types and HPC collective communication libraries such as NCCL, RCCL, UCC and MPI
Innovate and co-design novel model architectures for sustained scaling and hardware efficiency during training and inference
Benchmark, analyze, model and project the performance of AI workloads against a wide range of what-if scenarios and provide early input to the design of future hardware, models and runtime, giving crucial feedback to the architecture, compiler, kernel, modeling and runtime teams
Explore, co-design and productionize model compression techniques such as Quantization, Pruning, Distillation and Sparsity to improve training and inference efficiency
Collaborate with AI & Systems Co-design to guide Meta’s AI HW strategy

Research Scientist Intern, MSL Infra Kernels & Optimizations

Meta’s Meta SuperIntelligence Labs (MSL) Infra Kernels & Optimizations (K&O) tea...

Location

United States , Menlo Park

Salary:

7650.00 - 12134.00 USD / Month

Meta

Expiration Date

Until further notice

Requirements

Currently has, or is in the process of obtaining, a PhD degree in the field of Computer Science, Computer Vision, Generative AI, NLP, relevant technical field, or equivalent practical experience
Must obtain work authorization in country of employment at the time of hire, and maintain ongoing work authorization during employment
Specialized experience in one or more of the following areas: Accelerators/GPU architectures, High Performance Computing (HPC), Machine Learning Compilers, Training/Inference ML Systems, Model Compression, Communication Collectives, ML Kernels/Operator optimizations, Machine learning frameworks (e.g. PyTorch) and SW/HW co-design
Experience developing AI-System infrastructure or AI algorithms in C/C++ or Python

Job Responsibility

Explore, prototype and productionize highly optimized ML kernels to unlock full potential of current and future accelerators for Meta’s AI workloads. Open source SOTA implementations as applicable
Explore, co-design and optimize parallelisms, compute efficiency, distributed training/inference paradigms and algorithms to improve the scalability, efficiency and reliability of inference and large-scale training systems
Optimize inference and training communications performance at scale and investigate improvements to algorithms, tooling, and interfaces, working across multiple accelerator types and HPC collective communication libraries such as NCCL, RCCL, UCC and MPI
Innovate and co-design novel model architectures for sustained scaling and hardware efficiency during training and inference
Benchmark, analyze, model and project the performance of AI workloads against a wide range of what-if scenarios and provide early input to the design of future hardware, models and runtime, giving crucial feedback to the architecture, compiler, kernel, modeling and runtime teams
Explore, co-design and productionize model compression techniques such as Quantization, Pruning, Distillation and Sparsity to improve training and inference efficiency
Collaborate with AI & Systems Co-design to guide Meta’s AI HW strategy

New

Data Business Analyst, Power BI Specialist

Our client is a leading privately owned global asset manager with a range of suc...

Location

United Kingdom , London

Salary:

Not provided

Stephen Bell Associates

Expiration Date

Until further notice

Requirements

The ability to translate complex business concepts into scalable solutions
Hands-on PowerBI expertise including dashboard design, data modelling, advanced DAX and performance optimisation
An analytical mindset with excellent attention to detail and data integrity
Strong stakeholder engagement and communication skills
Experience delivering analytics solutions within asset management or the wider financial services industry
Exposure to data platforms and governance
The capacity to prioritise and work effectively across functions, regions and evolving priorities

Job Responsibility

Data analysis and requirements definition: partnering with stakeholders to understand needs, gather structured requirements and define KPIs, metrics and functional specifications. Define and document metrics and dictionaries to ensure consistency
PowerBI design and delivery: design, develop and maintain high-quality PowerBI dashboards aligned with governed products and user-friendly and intuitive visualisations. This includes developing measures (DAX), calculated fields and semantic models aligned with agreed business definitions
Data alignment, governance and improvement: work closely with data engineers to ensure data products and models support BI needs and ensure dashboards align with enterprise standards and governance frameworks. Promote best practice in dashboard design and data storytelling

Fulltime

New

Casual Early Years Educator

Joining as a casual colleague allows for a perfect work/life balance by setting ...

Location

United Kingdom , Albrighton

Salary:

12.75 - 13.80 GBP / Hour

RAFA Kidz

Expiration Date

Until further notice

Requirements

Genuine love of children
warmth
patience
great sense of humour
willingness to learn from colleagues to understand EYFS (Early Years Foundation Stage)
satisfactory, enhanced Disclosure & Barring Service (DBS) check on appointment and every three years thereafter

Job Responsibility

Being hands on and creative with messy and outdoor play activities
helping with meals and snack times
keeping records on children’s progress and daily activities and adventures
maintaining highest standards of safety, cleanliness and hygiene
meet and greet families and build great relationships

What we offer

Flexible working arrangement
potential to change to permanent contracts when roles become available

Parttime

Research Scientist Intern, PyTorch Compiler

Meta

Location:
United States , Menlo Park

Category:
IT - Software Development

Contract Type:
Not provided

Salary:

Job Description:

Job Responsibility:

Requirements:

Nice to have:

Additional Information:

Job Posted:
January 29, 2026

Looking for more opportunities? Search for other job offers that match your skills and interests.

Similar Jobs for Research Scientist Intern, PyTorch Compiler

Research Scientist Intern, PyTorch Framework Performance

Research Scientist Intern, PyTorch On-Device

Research Scientist Intern, AI & Compute Foundation - MTIA Software

Research Scientist Intern, PyTorch Distributed

Research Scientist Intern, MSL Infra Kernels & Optimizations

Research Scientist Intern, MSL Infra Kernels & Optimizations

Data Business Analyst, Power BI Specialist

Casual Early Years Educator

Research Scientist Intern, PyTorch Compiler

Meta

Location:United States , Menlo Park

Category:IT - Software Development

Contract Type:Not provided

Salary:

Job Description:

Job Responsibility:

Requirements:

Nice to have:

Additional Information:

Job Posted:January 29, 2026

Looking for more opportunities? Search for other job offers that match your skills and interests.

Similar Jobs for Research Scientist Intern, PyTorch Compiler

Research Scientist Intern, PyTorch Framework Performance

Research Scientist Intern, PyTorch On-Device

Research Scientist Intern, AI & Compute Foundation - MTIA Software

Research Scientist Intern, PyTorch Distributed

Research Scientist Intern, MSL Infra Kernels & Optimizations

Research Scientist Intern, MSL Infra Kernels & Optimizations

Data Business Analyst, Power BI Specialist

Casual Early Years Educator

Location:
United States , Menlo Park

Category:
IT - Software Development

Contract Type:
Not provided

Job Posted:
January 29, 2026