Research Scientist Intern, PyTorch On-Device (PhD) jobs in United States
cer-icon
Apply on Employer Site
company-logo

Meta · 12 hours ago

Research Scientist Intern, PyTorch On-Device (PhD)

Meta is a technology company that builds platforms to connect people and foster communities. They are seeking a Research Scientist Intern to develop and apply performance techniques for on-device AI, collaborating with users and developers to advance machine learning frameworks.

Computer Software
check
Comp. & Benefits

Responsibilities

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

Qualification

ML compilersSparsityQuantizationKernel developmentPerformance optimizationCross-functional communication

Required

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

Preferred

Experience working on other AI/ML optimized runtime stacks
Experience with performance optimization of machine learning models for on-device inference
Proven track record of achieving significant results as demonstrated by grants, fellowships, patents, as well as first-authored publications at leading workshops or conferences such as NeurIPS, MLSys, ASPLOS, PLDI, CGO, PACT, ICML, or similar
Experience working and communicating cross functionally in a team environment

Benefits

Benefits

Company

Meta's mission is to build the future of human connection and the technology that makes it possible.

Funding

Current Stage
Late Stage

Leadership Team

leader-logo
Kathryn Glickman
Director, CEO Communications
linkedin
leader-logo
Christine Lu
CTO Business Engineering NA
linkedin
Company data provided by crunchbase