Remote Research Engineer - AI Systems

Description:

• Seeking a highly motivated AI Systems Research Engineer specializing in Trainium, GPU kernels, and LLM systems optimization.
• Work on AI Systems, Compiler and Runtime Optimization, Distributed Training & Inference, GPU/Accelerator Kernel Development, and Large Language Model Infrastructure.

Requirements:

• Proficiency in AI programming languages such as 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 (e.g., PyTorch, Dynamo) and profiling tools (e.g., Nsight, ROCm Profiler).
• Strong problem-solving skills and ability to work in a collaborative, remote environment.
• Background in computer science, engineering, or a related field preferred.

Benefits:

• Be part of a visionary team redefining AI infrastructure and influencing the future of multi-silicon AI computing.
• Work on cutting-edge technologies solving frontier AI infrastructure problems.
• Collaborate with experts from leading institutions and tech companies.
• Competitive compensation with equity and a flexible, remote work environment.