Research Engineer

Research Engineer - AI Systems @ Yotta Labs - Back to Yotta Labs’s Job Listings Research Engineer - AI Systems Location United States; Canada; Hong Kong; Singapore Employment Type Full time Location Type Remote Department Engineering Location: Remote (Global) Type: Full-time 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 AI Systems Research Engineer specializing in Trainium, GPU kernels, and LLM systems optimization. You will work at the intersection of AI Systems, Compiler and Runtime Optimization, Distributed Training & Inference, GPU/Accelerator Kernel Development, and Large Language Model Infrastructure. Your work will directly impact the scalability and performance of AI applications deployed on our platform. 🎯 Responsibilities Design and implement high-performance kernels for Attention, MoE, GEMM, collective communication, and quantization. Optimize kernels for NVIDIA, AMD, and AWS Trainium. Develop custom operators and graph optimizations using Neuron SDK, PyTorch/XLA, Torch Dynamo, and Neuron Compiler. Improve performance of vLLM, SGLang, TensorRT-LLM, and custom inference runtimes. Design scalable distributed training and inference solutions across thousands of accelerators. Contribute to open-source projects, publish technical findings and engage with the developer community. ✅ Qualifications Proficiency in AI programming languages such as Python a