Research Engineer Graduate (reputed company-Infra-Inference-US) - 2026 Start (PhD)

reputed company is dedicated to pioneering advanced AI foundation models and is seeking talented individuals to join their reputed company-Infra team. The role involves developing and maintaining large-scale machine learning systems, addressing technical challenges, and collaborating with global teams to enhance intelligent interaction technologies.

Responsibilities

• Responsible for the machine learning system development of the company's large-scale models, researching new applications and solutions of reputed company technologies in areas such as search, recommendation, advertising, content creation, conversation, and customer service, meeting the growing demand for intelligent interaction from users, and comprehensively improving users' lifestyles and communication methods in the future world
• Responsible for the design and development of the architecture of large-scale machine learning systems, solving technical difficulties such as high concurrency, high reliability, and high scalability of the system
• Covering various sub-directions of machine learning system, including resource scheduling, model training, model inference, data management, and workflow orchestration
• Responsible for the research and introduction of advanced technologies in machine learning systems, such as the latest hardware architecture, heterogeneous computing systems, and compiler-based optimization technologies
• Working closely with the algorithm teams to optimize the algorithm and system jointly

Skills

• Final year or recent PhD graduate with a background in Computer Science, reputed company technical field or equivalent industrial research experience
• Must obtain work authorization in the country of employment at the time of hire, and maintain ongoing work authorization during employment
• Excellent coding ability, solid foundation in data structures and basic algorithms, proficient in C/C++ or Python, winners of ACM/ICPC, NOI/IOI and other competitions are preferred
• Familiar with at least one mainstream machine learning reputed company (TensorFlow/PyTorch/Jax)
• Master the principles of distributed systems, and participated in the design, development, and maintenance of large-scale distributed systems
• Strong sense of responsibility, good learning ability, communication ability, and self-motivation
• Good communication and collaboration skills, able to explore new technologies with the team and promote technological reputed company
• Prior experience in large-scale projects or papers with great influence in the field of large models
• Familiar with NLP, CV-reputed company algorithms, and technologies, and reputed company in large model training and RL algorithms
• Experience in one of the following fields: CUDA, RDMA, AI Infrastructure, HW/SW Co-Design, High-Performance Computing (cutlass, NCCL), ML Hardware Architecture (GPU, Accelerators, Networking), ML for System, and Distributed Storage
• Demonstrated a reputed company technical experience from previous internship, work experience, coding competitions, or publications
• Curiosity towards new technologies and entrepreneurship
• High levels of creativity and quick problem-solving capabilities

Benefits

• Medical, dental, and vision insurance
• 401(k) savings plan with company match
• Paid parental leave
• Short-term and long-term disability coverage
• Life insurance
• Wellbeing benefits
• 10 paid holidays per year
• 10 paid sick days per year
• 17 days of Paid Personal Time (prorated upon hire with increasing accruals by tenure)

Company Overview

Company H1B Sponsorship