Research Engineer Intern - AI Systems

Type: Internship

Company: Yotta Labs

Apply: careers@yottalabs.ai

🧠 About Yotta Labs

Yotta Labs is building the next generation multi-silicon AI cloud and runtime platform to power the world’s most demanding AI workloads. We enable training and inference across NVIDIA GPUs, AMD GPUs, and AWS Trainium, helping AI companies achieve the best performance and economics across heterogeneous hardware. Our mission is to provide high-performance AI computing and Model API services, enabling AI companies, research labs, and enterprises to train, deploy and integrate cutting-edge models at scale.

🛠️ Role Overview

We are seeking a highly motivated Research Engineer Intern to work on Trainium, GPU kernels, and LLM systems optimization. Over a 12–16 week internship, you will own a well-scoped project at the intersection of AI Systems, Compiler and Runtime Optimization, Distributed Training & Inference, GPU/Accelerator Kernel Development, and Large Language Model Infrastructure — taking it from design to working, profiled code running on real hardware. Your work will ship to production or open source and directly impact the performance of AI applications deployed on our platform. Strong interns receive return offers for full-time roles.

🎯 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.

✅ Qualifications