Publié il y a 4 mois
Research Engineer - AI Systems
@ Yotta Labs
Télétravail au Canada
En bref
Yotta Labs is hiring a full-time Research Engineer for AI Systems, fully remote and global. The role optimizes GPU and Trainium kernels, compilers and LLM inference runtimes across heterogeneous accelerators.
Responsabilités :
- Design high-performance kernels for Attention, MoE, GEMM and quantization
- Optimize kernels for NVIDIA, AMD and AWS Trainium
- Develop custom operators and graph optimizations
- Improve vLLM, SGLang and TensorRT-LLM inference runtimes
- Design distributed training and inference across thousands of accelerators
Résumé des exigences :
- Proficiency in Python and C++
- Deep GPU architecture and performance optimization knowledge
- Experience with CUDA, Triton, ROCm/HIP or AWS Neuron
- Familiarity with PyTorch, Dynamo and profiling tools
- Background in computer science or engineering preferred
Outils et compétences
Source et dates
Site carrière / système de candidatures · ashby
- Date publiée par l’employeur
- 2026-06-28
- Première détection par notre source
- 2026-10-11
- Mise à jour du dossier source
- 2026-10-11
Ces dates décrivent le dossier source, pas une vérification en direct du recrutement. Consultez l’offre originale.
Détails du poste
| Domaine | Développement logiciel |
| Heures par semaine | 40 (estimé) |
| Déplacements | avion : aucun · route : aucun |
| Milieu de travail | Domicile |
| Position | Surtout assis |
| Exigences physiques | physique faible · ordinateur élevé |
| Langues | English |
| Domaines d’études | Computer science, Engineering |
| Avantages mentionnés | Actions ou participation |
Résumé et détails rédigés par l’IA à partir de l’offre; l’offre de l’employeur fait foi.
Yotta Labs
Profil d’entreprise sauvegardé
Renseignements historiques provenant d’un profil The Jobs API sauvegardé (champs LinkedIn/Crunchbase). Ils ne sont pas vérifiés en temps réel; les chiffres peuvent avoir changé. Les valeurs manquantes restent inconnues.
Building the next generation multi-silicon AI cloud and runtime platform.
Description complète de l’entreprise
Yotta Labs is at the forefront of building the next generation multi-silicon AI cloud and runtime platform. We enable training and inference across NVIDIA GPUs, AMD GPUs, and AWS Trainium/Inferentia, helping AI companies achieve the best performance and economics across heterogeneous hardware
- Abonnés LinkedIn
- 741
- Personnes associées sur LinkedIn
- 7
- Bureaux déclarés
- Seattle, US
- Catégories d’activité
- GPU
- Cloud Computing
- Artificial Intelligence (AI)
- Autres noms enregistrés
- Yotta Labs
- Le profil indique une agence de recrutement
- Non
- Investissement total déclaré
- 0
La devise et la date d’observation ne sont pas fournies; ce n’est ni une valorisation actuelle ni une ronde de financement.
Yotta Labs is at the forefront of building the next generation multi-silicon AI cloud and runtime platform. We enable training and inference across NVIDIA GPUs, AMD GPUs, and AWS Trainium/Inferentia, helping AI companies achieve the best performance and economics across heterogeneous hardware
| Champ | Valeur |
|---|---|
| Secteur | Technology, Information and Internet |
| Taille | 2-10 employees |
| Siège social | Seattle |
| Type d’organisation | Public Company |
| Fondée en | 2024 |
| Offres sur Jobsy | 3 |
| Voir la page |
Location: Remote (Global)
Type: Full-time
Company: Yotta Labs
Apply: [email protected]
🧠 About Yotta Labs
Yotta Labs is building the next generation multi-silicon AI cloud and runtime platform to power the world’s most demanding AI workloads. We enable training and inference across NVIDIA GPUs, AMD GPUs, and AWS Trainium, helping AI companies achieve the best performance and economics across heterogeneous hardware. Our mission is to provide high-performance AI computing and Model API services, enabling AI companies, research labs, and enterprises to train, deploy and integrate cutting-edge models at scale.
🛠️ Role Overview
We are seeking a highly motivated AI Systems Research Engineer specializing in Trainium, GPU kernels, and LLM systems optimization. You will work at the intersection of AI Systems, Compiler and Runtime Optimization, Distributed Training & Inference, GPU/Accelerator Kernel Development, and Large Language Model Infrastructure. Your work will directly impact the scalability and performance of AI applications deployed on our platform.
🎯 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.
✅ Qualifications
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.
🌟 Preferred Experience
Contributions to open-source AI infra projects like vLLM, SGLang, PyTorch, or Triton.
Experience with with FlashAttention, PagedAttention, MoE, RLHF, or distributed AI systems.
Publications in top-tier conferences like MLSys, OSDI, SOSP, NSDI, SC, HPCA, or ISCA
🌐 Why Join Yotta Labs?
Be part of a visionary team aiming to redefine AI infrastructure and influence the future of multi-silicon AI computing.
Work on cutting-edge technologies that solves frontier AI infrastructure problems.
Collaborate with experts from leading institutions and tech companies.
Competitive compensation with equity. Enjoy a flexible, remote work environment that values innovation and autonomy.
📩 How to Apply
Interested candidates should apply directly or send their resume and a brief cover letter to [email protected]. Please include links to any relevant projects or contributions.
Journal
Visible seulement par vous. Conservé sur cet appareil, et dans votre compte une fois connecté.