Research Engineer Intern focused on AI Systems at Yotta Labs. Work on optimizing GPU kernels and LLM infrastructure in a remote setting.
Responsibilities
Implement and optimize compute kernels for Attention, GEMM, MoE, and quantization on NVIDIA, AMD, or AWS Trainium.
Build custom operators using CUDA, Triton, ROCm/HIP, or the Neuron SDK with PyTorch/XLA.
Profile and improve inference performance in vLLM, SGLang, and our custom runtimes — kernel fusion, scheduling, KV-cache and memory optimizations.
Build benchmarks, chase down performance regressions, and turn profiler traces into concrete speedups.
Ship code upstream to open-source AI infrastructure projects, with tests and documentation.
Requirements
Currently pursuing a BS, MS, or PhD in Computer Science, Computer Engineering, or a related field
Solid programming skills in Python and familiarity with C++
Understanding of GPU/accelerator architecture fundamentals (memory hierarchy, parallelism, occupancy) from coursework, research, or projects
Experience writing CUDA, Triton, ROCm/HIP, or Neuron kernels — class projects and personal projects count
Strong understanding of AI frameworks (e.g., PyTorch, Dynamo, LMCache), model architectures and profiling tools (e.g. Nsight, ROCm Profiler, or Neuron Profiler)
Strong problem-solving skills and the ability to work independently in a collaborative, remote environment.
Benefits
Competitive internship compensation
Flexible remote work environment
Direct mentorship from engineers from leading institutions and tech companies
Fast path to a full-time return offer for top performers
Clinical Systems Engineering Co - op analyzing reliability data for Profound Medical’s MRI - guided robotic ultrasound therapy system. Developing Python/MATLAB tools and supporting MRI and lab testing.
Financial Systems Analyst supporting Ledcor’s construction financial systems in Edmonton or Calgary. Maintaining JD Edwards, reporting, data integrity, troubleshooting, and system improvements.
Genesys Systems Engineer supporting routing, IVR, and omnichannel contact center technology. Helping Forward Financing, a fintech lender, optimize customer and agent experiences.
Systems Engineer II developing and validating spaceborne optical instruments at Honeywell Aerospace. Defining requirements, architectures, and verification strategies for aerospace and defense missions.
Principal Systems Architect defining autonomous - vehicle platform architecture at General Motors. Leading safety - critical systems strategy, requirements, verification, and technical direction across vehicle programs.
People Operations Systems Analyst optimizing HRIS platforms and workflows for Medisca, a global healthcare solutions company. Improving data accuracy, automation, reporting, and employee experience.
People Operations Systems Analyst optimizing HRIS platforms and workflows for Medisca, a global healthcare solutions company. Driving system implementations, automation, data accuracy, and employee experience improvements.
Concepteur principal dirigeant la conception de systèmes BAS/CVAC chez Automated Logic. Développement, programmation, essais et mentorat pour des solutions intelligentes d’automatisation des bâtiments.
Senior Systems Engineer integrating LiDAR, camera, and edge - computing systems for Invision AI’s intelligent infrastructure products. Developing C++ software, validating hardware, and supporting customer deployments.