Research Engineer Intern – AI Systems
Employment Information
Report this job
Job expired or something wrong with this job?
Job Description
Research Engineer Intern focused on AI Systems at Yotta Labs. Work on optimizing GPU kernels and LLM infrastructure in a remote setting.
Responsibilities:
- Implement and optimize compute kernels for Attention, GEMM, MoE, and quantization on NVIDIA, AMD, or AWS Trainium.
- Build custom operators using CUDA, Triton, ROCm/HIP, or the Neuron SDK with PyTorch/XLA.
- Profile and improve inference performance in vLLM, SGLang, and our custom runtimes — kernel fusion, scheduling, KV-cache and memory optimizations.
- Build benchmarks, chase down performance regressions, and turn profiler traces into concrete speedups.
- Ship code upstream to open-source AI infrastructure projects, with tests and documentation.
Requirements:
- Currently pursuing a BS, MS, or PhD in Computer Science, Computer Engineering, or a related field
- Solid programming skills in Python and familiarity with C++
- Understanding of GPU/accelerator architecture fundamentals (memory hierarchy, parallelism, occupancy) from coursework, research, or projects
- Experience writing CUDA, Triton, ROCm/HIP, or Neuron kernels — class projects and personal projects count
- Strong understanding of AI frameworks (e.g., PyTorch, Dynamo, LMCache), model architectures and profiling tools (e.g. Nsight, ROCm Profiler, or Neuron Profiler)
- Strong problem-solving skills and the ability to work independently in a collaborative, remote environment.
Benefits:
- Competitive internship compensation
- Flexible remote work environment
- Direct mentorship from engineers from leading institutions and tech companies
- Fast path to a full-time return offer for top performers
Similar Jobs
System Safety Engineer
System Safety Engineer ensuring vehicle variants meet safety regulations and standards. Collaborating on documentation, verification, and validation processes in the automotive sector.
Senior Core Systems Engineer
Senior Core Systems Engineer developing features for Unreal Engine at Disney Digital Entertainment. Leveraging expertise in C++, Verse, and Python technology for game engines.
Senior Systems Engineer – Five9
Senior Five9 Systems Engineer managing cloud-based contact center platforms. Delivering value through configuration, integration, and operational resilience.
Trainer / Instructor, Vocational Trainer for Retraining IT Specialist (System Integration)
Trainer for IT System Integration at WBS Training, focusing on competence development and teaching. Conducts lessons remotely utilizing advanced teaching methods.
Trainer – Instructor for Retraining IT Specialist – System Integration
Trainer in IT System Integration delivering courses and supporting learners in competency development. This role involves remote teaching and preparation of educational materials.
Consulting Technologist, Manufacturing Systems Engineering
Consulting Engineer leading projects with high impact on the business at GE Aerospace. Responsible for managing complex issues and contributing to overall business strategy in manufacturing systems engineering.
Software Engineer, Autonomous Systems
Software Engineer building software for autonomous haulers at Humble Robotics. Working on vehicle systems and collaborating with a team of industry veterans.
Senior Refrigeration Systems Engineer
Senior Refrigeration Systems Engineer overseeing design, optimization, and reliability of facility systems at US Foods. Collaborating with various teams to enhance operational performance in a fully remote capacity
Senior System Engineer
Senior System Engineer at Rochester Regional Health architecting and designing server solutions. Providing complex support and maintaining high standards in system performance and engineering knowledge.
Systems Engineer – Planificación Batch
Systems Engineer at Inetum focusing on Batch planning, UNIX environment, and Control-M. Collaborating on innovative projects and contributing to digital transformation efforts.
Senior Business Systems Analyst – 12-month Contract
Senior Business Systems Analyst at The Salvation Army optimizing enterprise platforms like Salesforce and ServiceNow. Partnering with teams to translate business needs into functional requirements.
Clinical Systems Analyst
Clinical Systems Analyst focusing on laboratory information management systems for healthcare clients. Collaborating with clients to enhance clinical procedures and support technology implementation.
RHEL Systems Engineer
Senior RHEL Engineer responsible for OS and container platform management in energy-management deployments. Building tailored RHEL images and hardening containerized workloads for data centers.
Systems Engineer
Seeking Systems Engineer to design, implement, and maintain systems for business operations. Collaborate with teams to ensure system stability, performance, and innovation.
Full Stack Engineer, AI Systems
Full Stack Engineer developing AI-native product features for a smart assistant application. Focused on building reliable, real-world workflows and system integrations across frontend and backend.
Senior Principal Systems Engineer – MilAv OPAS
Senior Principal Systems Engineer at RTX developing avionics systems for aerospace and defense. Managing technical performance and leading system development for innovative aerospace solutions.
Systems Engineering Manager – HPE Networking, Mid-Telco Accounts
Systems Engineering Manager leading North America team for telecommunications at Hewlett Packard Enterprise. Building high-performing teams and engaging with strategic customers for technical excellence.
Systems Engineer – Channel SecOps SME
Systems Engineer enabling partners to effectively position, sell, and deliver SecOps solutions. Leading technical enablement and collaboration within Fortinet’s partner ecosystem in Italy.
Embedded Software Engineer, Tech Lead
Embedded Software Engineer, Tech Lead at Astera Labs focuses on firmware development for analog and Ethernet devices. Collaborating with a small team on high-speed networking and automation solutions.
Inventory Systems Analyst
Inventory Systems Analyst at Fellers, managing ERP data and supporting supply chain operations. Collaborating with teams to ensure product data accuracy and operational integrity across various departments.