About this role
Research Engineer - AI Systems at Yotta Labs. Location: United States or Hong Kong or Canada or Singapore. Role: designing kernels, optimizing kernels, developing operators Requirements: Proficiency in Python and C++; deep GPU architecture and optimization knowledge; experience with CUDA, Triton, ROCm/HIP or AWS Neuron; familiarity with PyTorch and profiling tools; degree in CS/engineering preferred. Category: Engineering Seniority: Senior Level Tools: Python, C++, CUDA, Triton, ROCm/HIP, AWS Neuron, Neuron SDK, PyTorch/XLA, Torch Dynamo, Neuron Compiler, vLLM, SGLang, TensorRT-LLM, LMCache, Nsight, ROCm Profiler, Neuron Profiler, FlashAttention, PagedAttention Commitment: Full Time Workplace: Remote Languages: English