Research Engineer - AI Systems
Skills
Computer ScienceAmazon Web ServicesDistributed TrainingEngineeringLarge Language ModelsNeuronsOptimization
What the job involves
The main requirements, responsibilities and hiring steps.
Requirements
- Proficiency in Python and C++
- Deep understanding of GPU architecture and performance optimization
- Experience with CUDA Triton ROCm HIP or AWS Neuron
- Strong understanding of AI frameworks model architectures and profiling tools
- Strong problem-solving skills and ability to work collaboratively in a remote environment
- Background in computer science engineering or related field preferred
Nice to have
- Open-source contributor
- Systems thinker
- Research-oriented
- Collaborative
- Independent
Day to day
- Design and implement high-performance kernels for attention MoE GEMM collective communication and quantization
- Optimize AI workloads across NVIDIA AMD and AWS Trainium hardware using advanced compiler and runtime techniques
- Build scalable distributed training and inference systems and improve open-source LLM runtimes for better performance
Hiring process
- Apply directly
- Send resume and brief cover letter to careers@yottalabs.ai
- Include links to relevant projects or contributions