Research Scientist Intern, PyTorch On-Device (PhD)

This internship is based in Bellevue, WA with additional locations available.

Compensation:

$7,650/month to $12,134/month + 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.

Minimum Qualifications:

• 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.

Preferred Qualifications:

• Experience working on other AI/ML optimized runtime stacks.
• Experience with performance optimization of machine learning models for on-device inference.
• Intent to return to degree program after the completion of the internship/co-op.
• Proven track record of achieving significant results as demonstrated by grants, fellowships, patents, as well as first-authored publications at leading workshops or conferences.
• Experience working and communicating cross-functionally in a team environment.