Yotta Labs

Systems Engineer

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

Research Engineer – AI Systems