Research Engineer, ML Systems (All Industry Levels)
Tasks
- Build distributed RLHF stack
- Develop Triton kernels
- Develop multimodal model training and inference systems
- Develop routing algorithms for serving
- Optimize GPU training performance
- Optimize inference runtime
- Train and distill LLMs
- Tune kernels for models and hardware
- Write production system code
Perks/Benefits
- N/A
Skills/Tech-stack
CUDA | CUDA kernels | Cloud | Cutlass | DeepSpeed | Distributed Systems | Distributed Training | Docker | FlashAttention | Kubernetes | LLM Inference | Machine Learning | Megatron | PyTorch | PyTorch distributed | RLHF | Reinforcement Learning | Transformers | Triton | VLLM
Education
Regions
Countries
States
Cities
Related jobs
-
Senior-level Full TimeUnited States2h ago
-
Senior-level Full TimeUnited States2h ago
-
Computational Biologist - Protein Engineering USD 150K-250KAWS | Amazon Web Services | CUDA | Conda | Deep learningRelocation supportEntry-level Full TimeSan Francisco, CA, US3h ago
-
Entry-level Full TimeSan Francisco, CA, US3h ago
-
Applied Math Libraries Engineer (SPU) USD 120K-170KArithmetic intensity | Atomics | BLAS | Bandwidth) | C#Entry-level Full TimeRedwood City3h ago
-
Kernel Engineer — Scientific Computing (SPU) USD 120K-170KAtomic operations | C++ | CUDA | Concurrency | DebuggingEntry-level Full TimeRedwood City3h ago
-
Senior Google Cloud Data Engineer USD 170K-240KAlerting | Anomaly Detection | Apache Beam | Apache PubSub | Batch Data ProcessingCareer advancement | Dental insurance | Health savings account | Hybrid work options | Medical insuranceSenior-level Full TimeNew York City5h ago
-
Machine Learning Engineer Intern (E-Commerce Knowledge Graph - CV/Multimodal/NLP) - 2026 Summer (BS/MS) USD 122K-243KEntity Linking | Graph Modeling | Knowledge graphs | Language Processing | Machine LearningDevelopment workshops | Social eventsEntry-level InternshipSeattle, Washington, United States6h ago
-
Senior Software Engineer, AI Coding Tools USD 244K-588KDeep learning | Fine Tuning | GPU | Inference acceleration | Language ModelsSenior-level Full TimeSan Jose, California, United States6h ago
-
Machine Learning Engineer Intern (E-commerce-Conversational AI) - 2026 Summer/Fall (PhD) USD 129K-246KChatbot Development | Information Retrieval | Language Models | Language Processing | Large Language ModelsEntry-level InternshipSeattle, Washington, United States6h ago
-
Automated Orchestration | Backup and Restore | C# | C++ | CXLSenior-level Full TimeSeattle, Washington, United States6h ago
-
Checkpointing | Data-Driven Optimization | Data-driven | Distributed Training | Fault ToleranceMid-level Full TimeSan Jose, California, United States6h ago
-
Machine Learning Engineer Intern (E-commerce-Supply Chain & Logistics) - 2026 Summer (MS) USD 136K-237KClassification | Data Mining | Deep learning | Forecasting | Knowledge graphsEntry-level InternshipSeattle, Washington, United States6h ago
-
Software Engineer, Generative AI, Workspace USD 147K-211KC++ | Distributed Systems | Generative AI | Information Retrieval | Integration TestingBenefits | Bonus | EquityMid-level Full TimeBoulder, CO, USA7h ago
-
Staff Software Engineer, Machine Learning, Google Chat USD 207K-300KAgentic Workflows | Caching | Cloud Spanner | Continuous Delivery | Continuous integrationSenior-level Full TimeSunnyvale, CA, USA7h ago
-
Software Engineer III, Database Internals AlloyDB USD 147K-211KACID | C# | C++ | CAP Theorem | Compiler TheoryEntry-level Full TimeSunnyvale, CA, USA7h ago
-
Software Engineer III, AI/ML Computer Vision, AR USD 147K-211KC++ | Computer Vision | Data Processing | Data Structures | Data Structures and AlgorithmsSenior-level Full TimeSan José, CA, USA7h ago
-
Agentic AI | C plus plus | C# | Cloud services | Data ProcessingMid-level Full TimeSan Francisco, CA, USA7h ago
-
Supply Chain Data Engineer Ii USD 94K-118KDBT | Data Governance | Data Modeling | Data Pipelines | Data Quality401k | Disability insurance | Employee stock purchase plan | Health insurance | Life insuranceMid-level Full TimeWayne, PA, US, 1908712h ago
-
AI/ML Engineer 2 USD 101K-165KAI Agents | API Development | AWS | Azure | CI/CDDisability insurance | Family leave | Flexible spending accounts | Life and AD D Insurance | Medical/Dental/Vision insuranceSenior-level Full TimePhiladelphia, PA, US, 1910312h ago
-
Staff AI/ML Engineer USD 240K-270KAWS | Agentic Workflows | Cloud platform | Data Curation | Deep learning401k | Commuter benefits | Dog-friendly office | Equity | FSA benefitsSenior-level Full TimeSan Francisco, CA12h ago
-
Senior Analytics Engineer, RegTech USD 150K-180KAWS | Acceptance Testing | Agile | Airflow | DBT401k match | Company-provided phone | Extended leave | Health insurance | Observed holidaysSenior-level Full TimeNew York, New York, United States15h ago
-
Applied AI ML Engineer-Vice President USD 150K-210KAWS Bedrock | AWS SageMaker | Amazon EKS | AutoPrompt | DDPBackup childcare | Financial coaching | Health care coverage | Mental health support | On-site health and wellness centersExecutive-level Full TimeNew York, NY, United States17h ago
-
Applied AI ML Engineer-Senior Associate USD 175K-210KAWS Bedrock | AWS SageMaker | Amazon EKS | Containerization | Data PreprocessingSenior-level Full TimeNew York, NY, United States17h ago
-
Senior-level Full Time142019-NC-300 South Brevard, Charlotte, United States18h ago