Research Engineer – AI Systems
Employment Information
Report this job
Job expired or something wrong with this job?
Job Description
AI Systems Research Engineer at Yotta Labs focusing on optimizing AI systems and GPU architecture for multi-silicon computing. Join a visionary team rethinking AI infrastructure through innovation and collaboration.
Responsibilities:
- 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.
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, LMCache), model architectures and profiling tools (e.g. Nsight, ROCm Profiler, or Neuron Profiler)
- Strong problem-solving skills and the ability to work in a collaborative, remote environment
- A background in computer science, engineering, or a related field is preferred
Benefits:
- Competitive compensation with equity
- Flexible, remote work environment that values innovation and autonomy
Similar Jobs
Knowledge Architect – Learning Systems Lead
Leader in knowledge management and learning systems at a technology company. Designing programs and resources to enhance organizational learning and performance.
People Systems Analyst
People Systems Analyst managing HR technology operations at Chainlink Labs. Supporting systems, documentation, and continuous improvement within a decentralized finance environment.
IT Systems Implementation Engineer
IT Systems Implementation Engineer serving as technical delivery owner on integration and AI projects. Leading discovery sessions, managing implementation, and collaborating with Subscriber IT teams.
Platform Engineer – AI Workflow Systems
Platform Engineer designing and building core systems for AI workflow automation at BJAK. Collaborating with teams in Malaysia while fully remote in China.
IT Business Systems Engineer – Web API
Business Systems Engineer translating business requirements into technical specifications. Supporting e-commerce platform and engaging with IT and business partners for effective solutions.
Exposure Intelligence Analyst – Infrastructure & Systems, Servers, Containers, K8s, OS
Exposure Intelligence Analyst managing infrastructure security challenges at Allstate. Identifying and validating exposure risks across servers, containers, and OS platforms with a focus on mitigation.
Senior Systems Developer
Senior Systems Developer on implementation team creating customized solutions for supply chain needs. Collaborating with clients to map processes and deliver high-quality systems.
Software Engineer, Embedded/Network Systems I
Software Engineer developing embedded and network systems for Cisco. Collaborating on hardware-software integration and low-level networking technologies.
Integrated Eligibility Systems Analyst
IES Analyst supporting Integrated Eligibility Systems projects in the public sector by performing testing and managing defects. Preferred candidates from Lake Mary, FL with relevant experience.
GTM Systems Engineer
GTM Systems Engineer focusing on Salesforce and HubSpot integration while building agentic workflows. Work at LiveKit to enhance revenue intelligence and operational efficiency.
Systems Analyst
Systems Analyst involved in building a new digital lending ecosystem for agricultural finance. Collaborating with teams to ensure technology solutions meet customer needs and align with business goals.
Systems Engineer
Systems Engineer providing remote technical support for client networks and servers in IT support at Thrive. Responsibilities include troubleshooting and managing customer service incidents with a focus on world-class support.
Senior Systems Engineer
Senior Systems Engineer working with customer environments across cloud, on-prem, and hybrid infrastructures. Engaging in high-impact responsibilities in architecture, security, and automation.
Senior Systems Analyst
Senior Systems Analyst supporting Salesforce solutions for healthcare organization. Collaborating with various teams to enhance CRM processes and integrations.
System Engineer, Customer Services
System Engineer Customer Services providing 1st to 3rd level IT support. Engaging with clients to ensure optimal IT performance while maintaining service quality and reliability.
Expert System Engineer
Expert System Engineer responsible for architecture and integration solutions for AT&T. Leading system design, technical leadership, and performance analysis for high-performance platforms.
Senior Systems Analyst – Accounting Operations
Senior Systems Analyst optimizing scalable, well-controlled Oracle Cloud ERP accounting operations for finance teams. Contributing to continuous improvements and ensuring financial integrity through system analysis.
AI Systems Engineer – Forward-Deployed Builder
AI Systems Engineer involved in building complex systems and products at a fast-paced AI company. A role focused on deconstructing workflows, building agentic systems, with autonomy and ownership.
Systems Engineer
Systems Engineer designing and developing software solutions for state-of-the-art safety-critical systems in Avionics and Space Exploration. Join a team with a strong history of stability and low turnover.
Systems Engineer – Mechatronics
Systems Engineer in Mechatronics group at Capgemini Engineering. Leading collaboration for new product development and improvements with a focus on safety and compliance.