2026 Summer Intern - Computational Sciences Center of Excellence - AI systems performance engineering jobs in United States
cer-icon
Apply on Employer Site
company-logo

Roche · 3 weeks ago

2026 Summer Intern - Computational Sciences Center of Excellence - AI systems performance engineering

Roche is a leading healthcare company committed to innovation and advancing science. They are seeking a PhD/Master's student to optimize GPU-accelerated workloads and improve AI-driven discovery by collaborating with machine learning scientists and engineers.

BiotechnologyHealth CareHealth DiagnosticsOncologyPharmaceuticalPrecision Medicine
check
Comp. & Benefits
check
H1B Sponsor Likelynote

Responsibilities

Assist in optimizing GPU-accelerated workloads, with a focus on throughput, latency, and memory efficiency
Profile and analyze machine learning training and inference workloads to identify performance bottlenecks, and apply optimizations at the graph, kernel, and system levels (e.g., PyTorch Profiler, NVIDIA Nsight)
Design and evaluate performance-aware algorithms that scale on multi-node clusters
Develop and optimize high-performance GPU kernels, and make trade offs to maximize hardware utilization
Develop a deep understanding of hardware features and performance characteristics to optimize large-scale AI and scientific computing workloads
Support benchmarking and performance testing efforts for AI systems at scale

Qualification

C/C++ programmingPython programmingCUDA optimizationGPU architecture understandingDistributed systems experienceParallel computing techniquesPassion for systemsBiological dataCommunication skills

Required

Must be pursuing a PhD/Master's (enrolled student)
Computer Sciences, Artificial Intelligence, Computational Sciences or a related field with a focus on machine learning systems, parallel computing, compilers or similar
Programming proficiency in C/C++ and Python
Experience writing and optimizing scientific computing kernels with CUDA or similar
Understanding of GPU or other accelerator architectures
A collaborative mindset and enthusiasm for bridging Computer Systems, Machine Learning Engineering and Biology

Preferred

Familiarity with compilers for scientific computing workloads (MLIR, TVM, etc.)
Familiarity with foundation model architectures (including but not limited to LLMs), training infrastructure and inference on HPC or cloud environments
Experience with distributed systems and parallel computing techniques, including data, model, and pipeline parallelism and checkpointing
Passionate about understanding and unpacking systems at a low level
Familiarity with biological data and drug development is helpful, but not required
Complements our culture and the standards that guide our daily behavior & decisions: Integrity, Courage, and Passion

Benefits

Paid holiday time off benefits

Company

Roche is a pharmaceutical and diagnostics company that offers medicines and diagnostic tests for various medical conditions and diseases.

H1B Sponsorship

Roche has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (12)
2024 (9)
2023 (6)
2022 (2)
2021 (2)

Funding

Current Stage
Public Company
Total Funding
$7.79B
Key Investors
SoftBankSCALE AINovartis
2021-08-04Post Ipo Equity· $5B
2020-12-07IPO
2020-05-06Post Ipo Equity· $0.5M

Leadership Team

leader-logo
Alan Hippe
Member of the Executive Board - Group CFO
linkedin
leader-logo
Christine Bakan
Global Head and VP of Software and Bioinformatics, Next Generation Sequencing
linkedin
Company data provided by crunchbase