Research Intern, Inference (Summer 2027)

Togetherai; San Francisco (CA); On-site; USD 80,000 - 96,000; Full time

Benefits offered by this job

Housing stipend

Competitive benefits

Job summary

Together AI, the AI Native Cloud, invites ambitious students to join the Inference Research team for an internship focused on distributed inference, compiler-aware optimization, and novel inference-time strategies. You will co-design cross-layer optimizations across models, systems, and hardware.

Preferred candidates are in final-year BS/MS/PhD programs, with strong ML/DL fundamentals, Python skills, and familiarity with PyTorch or JAX. Housing stipends and competitive benefits are provided.

Qualifications

• Currently pursuing final year in CS/EE or related field.
• Strong ML and DL fundamentals with Python proficiency.
• Experience with PyTorch or JAX and Transformer concepts is preferred.

Responsibilities

• Design and conduct rigorous experiments to validate hypotheses.
• Communicate plans, progress, and results to the team.
• Document findings in publications or blogs.

Skills

Machine Learning basics

Deep Learning fundamentals

Python programming

Transformer architectures

Education

Final year Bachelor's/Master's/Ph.D. in CS/EE

Tools

PyTorch

JAX

CUDA

Job description

About The Role

The Inference Research team is dedicated to building the next generation of efficient, scalable, and reliable serving systems for large foundation models, directly contributing to the mission of advancing open and transparent AI. Our work operates at the critical intersection of cutting-edge model architectures, high-performance systems engineering, and deep hardware optimization. We focus on co-designing software, algorithms, and models to significantly lower the cost and latency of modern AI systems.

As a research intern, you will dive into the complexities of distributed inference, compiler-aware optimization, and novel inference-time computation strategies (such as speculative decoding and phase-aware execution). You will be tasked with co-designing and implementing cross-layer optimizations across models, systems, and hardware, with a focus on areas like KV cache design and large-scale serving architectures.

Projects aim to unlock unprecedented performance and scale for foundation models, enabling faster serving, larger model deployment (e.g., Mixture-of-Experts), and robust, reproducible evaluation under realistic serving workloads.

Responsibilities

• Design and conduct rigorous experiments to validate hypotheses
• Communicate the plans, progress, and results of projects to the broader team
• Document findings in scientific publications and blog posts

Requirements

• Currently pursuing a final year of Bachelor's, Master's, or Ph.D. degree in Computer Science, Electrical Engineering, or a related field
• Strong knowledge of Machine Learning and Deep Learning fundamentals
• Experience with deep learning frameworks (PyTorch, JAX, etc.)
• Strong programming skills in Python
• Familiarity with Transformer architectures and recent developments in foundation models

Preferred Qualifications

• Prior research experience in foundation models, efficient machine learning, or ML systems.
• Publications at leading conferences in machine learning or systems (i.e., MLSys, ICLR).
• Experience with CUDA programming (for kernel development)
• Understanding of model optimization techniques and hardware acceleration approaches
• Contributions to open-source machine learning projects

About Together AI

Together AI, the AI Native Cloud, is purpose-built for AI engineers. AI application developers get high-performance inference that scales reliably, fine-tuning and reinforcement learning for creating frontier-level specialized models, and pre‑training at massive scale for fully custom intelligence, all around a marketplace of leading open models that teams can run, adapt, and own. Trusted by Cursor, Decagon, ElevenLabs, Salesforce, and Zoom, Together serves 400+ trillion tokens a month.

Internship Program Details

Our internship program runs 12 to 14 weeks, giving you the opportunity to work alongside industry-leading engineers and researchers across multiple teams. This cohort's internship dates span either May 17th to August 6th or June 14th to September 3rd.

Compensation

We offer competitive compensation, housing stipends, and other competitive benefits. The estimated US hourly rate for this role is $58 to $70 an hour. Our hourly rates are determined by location, level and role. Individual compensation will be determined by experience, skills, and job-related knowledge.

Equal Opportunity

Together AI is an Equal Opportunity Employer and is proud to offer equal employment opportunity to everyone regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity, veteran status, and more.

Please see our privacy policy at https://www.together.ai/privacy