Research Scientist Intern, Systems ML - SW/HW Co-Design - CHIPs (PhD)

Meta Platforms, Inc.•Menlo Park, CA

50d

About The Position

AI System SW/HW Co-design team's mission is to explore, develop and help productize high-performance software and hardware technologies for AI at datacenter scale. We achieve this via concurrent design and optimization of many aspects of the system such as models, algorithms, numerics, performance and AI hardware including compute, networking and storage. In essence, we drive the AI HW roadmap at Meta and ensure our existing and future AI workloads and software are well optimized and suited for the hardware infrastructure. Meta is seeking Research Scientist Interns to join our AI & Systems Co-Design CHIPs team to drive the definition of our next-generation compute and storage architectures. In this role, you will work cross-functionally with internal software and platforms engineering teams to understand the workloads and infrastructure requirements. You will drive technology path-finding, roadmap definition and co-design activities to deliver new capabilities and efficient systems for our fleet. You will also work with external industry partners to influence their roadmaps and build the best products for Meta's Infrastructure. Join our team and help shape one of the largest infrastructure footprints which powers Meta's applications used by billions of people across the globe. Our team at Meta AI offers twelve (12) to sixteen (16) weeks long internships and we have various start dates throughout the year. To learn more about our research, visit https://ai.facebook.com.

Requirements

Currently has, or is in the process of obtaining, PhD degree in the field of Computer Science or a related STEM field
Experience with hardware architecture, compute technologies and/or storage systems
Must obtain work authorization in the country of employment at the time of hire, and maintain ongoing work authorization during employment

Nice To Haves

Track record of achieving results as demonstrated by grants, fellowships, patents, as well as first-authored publications at leading workshops or conferences such as MICRO, ISCA, HPCA, ASPLOS, ATC, SOSP, OSDI, MLSys or similar
Architectural understanding of CPU, GPU, Accelerators, Networking, Flash/HDD Storage systems
Some experience with large-scale infrastructure, distributed systems, full stack analysis of server applications
Experience or knowledge in developing and debugging in C/C++, Python and/or PyTorch
Experience driving original scholarship in collaboration with a team
Experience leading a team in solving analytical problems using quantitative approaches
Experience in theoretical and empirical research and for answering questions with research
Experience communicating research for public audiences of peers
Experience working and communicating cross functionally in a team environment
Demonstrated software engineer experience via an internship, work experience, coding competitions, or widely used contributions in open source repositories (e.g. GitHub)
Intent to return to degree-program after the completion of the internship/co-op

Responsibilities

Develop tools and methodologies for large scale workload analysis and extract representative benchmarks (in C++/Python/Hack) to drive early evaluation of upcoming platforms.
Analyze evolving Meta workload trends and business needs to derive requirements for future offerings. Apply in depth knowledge of how the AI/ML systems interact with the compute and storage systems around.
Utilize extensive understanding of CPUs (x86/ARM), Flash/HDD storage systems, networking, and GPUs to identify bottlenecks and enhance product/service efficiency. Collaborate closely with software developers to re-architect services, improve codebase through algorithm redesign, reduce resource consumption, and identify hardware/software co-design opportunities.
Identify industry trends, analyze emerging technologies and disruptive paradigms. Conduct prototyping exercises to quantify the value proposition for Meta and develop adoption plans. Influence vendor hardware roadmap and broader ecosystem to align with Meta's roadmap requirements.
Work with Software Services, Product Engineering, and Infrastructure Engineering teams to find the optimal way to deliver the hardware roadmap into production and drive adoption.

Stand Out From the Crowd

Upload your resume and get instant feedback on how well it matches this job.

Upload and Match Resume