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
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
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
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 CompanyTotal Funding
$7.79BKey Investors
SoftBankSCALE AINovartis
2021-08-04Post Ipo Equity· $5B
2020-12-07IPO
2020-05-06Post Ipo Equity· $0.5M
Leadership Team
Recent News
2026-01-07
2026-01-07
Company data provided by crunchbase