Staff AI Engineer
Tasks
- Customize inference frameworks for production requirements
- Design inference serving systems for transformer and multimodal models
- Drive inference engineering practices across the team
- Implement inference optimizations for latency throughput and cost
- Monitor latency SLO and respond to incidents
- Own end to end model deployment and serving APIs
- Translate model architecture changes into inference efficient implementations
- Write and profile CUDA kernels and custom ops
Perks/Benefits
- N/A
Skills/Tech-stack
C plus plus | CUDA | CUDA kernels | Continuous batching | FP8 | GPU Programming | INT4) | INT8 | KV cache | Language Models | Large Language Models | Latency SLO | Memory Optimization | Multimodal Models | NVIDIA Nsight | ONNX Runtime | Prefill Decode | Prefill decode disaggregation | Python | Quantization | SGLang | Speculative decoding | TensorRT-LLM | Throughput Optimization | Transformer Models | VLLM
Education
Bachelor of Engineering | Bachelor of Science | Master of Science | PhD
Roles
AI | AI Engineer | Engineer | Learning Engineer | Machine Learning Engineer
Related jobs
-
Featured Feat. Principal Knowledge & Data Architect USD 174K-284KAWS Neptune | Canonicalization | Chunking | Cypher | DBTBenefits including health and wellness programs | Health, wellness, and retirement plansSenior-level Full TimeHeadquarters - Chevy Chase, MD R18d ago
-
Senior Data Engineer - Modern Data & AI Platforms USD 156K-204KAWS | Cortex | DBT | Data Governance | Data IngestionRemote workSenior-level Full TimeKansas City, MO, United States6h ago
-
Staff Software Engineer - AI Platform USD 175K-287KApache Beam | Apache Flink | Apache Spark | C# | C++Health and wellness programs | Time awaySenior-level Full TimeMountain View, CA, United States7h ago
-
Senior Software Engineer - AI Platform USD 144K-236KApache Beam | C# | C++ | CUDA | CUDA optimizationHealth and wellness programs | Time offSenior-level Full TimeMountain View, CA, United States7h ago
-
Senior Embedded Software Engineer USD 125K-160KAMD Xilinx | AMD Xilinx Devices | ARM Cortex | C# | Digital Signal401k match | Dental insurance | Employee assistance program | Every other Friday off | FSASenior-level Full TimeCamarillo, CA, United States11h ago
-
AI Engineer III USD 120K-140KAgent systems | Agentic AI | Amazon S3 | Amazon Web Services | Apache Airflow401k matching | Dental insurance | Employee assistance program | Flexible PTO | Life insuranceSenior-level Full TimeSan Antonio, TX, United States11h ago
-
AI/ML Systems Engineer USD 135K-210KAI Assurance | AI Governance | AI machine learning | Agentic Workflows | Anomaly DetectionSenior-level Full TimeDayton, OH, USA12h ago
-
Data Engineer USD 114K-160KCribl | Data Modeling | Data Warehouse | Data analytics | Data pipeline401k match | Holiday pay | Medical, dental & vision coverage | Paid time offMid-level Full TimeHuntsville, AL, United States12h ago
-
20454 - Analytics Engineer x 2 GBP 40K-42KAzure Data | Azure Data Factory | Azure Synapse | Data Factory | Data ModelingFlexible working | Hybrid working | Training and coachingEntry-level Full TimeNational, National14h ago
-
Senior Data Engineer USD 92K-135KAWS | Azure Data | Azure Data Factory | Change Data Capture | Cloud Security401k matching | Lifecycle benefits | Loan forgiveness | Tuition reimbursementSenior-level Full TimeChicago, IL, United States14h ago
-
Airflow | Apache Spark | CI/CD | Data Governance | Data LakehouseSenior-level ContractAustin, United States14h ago
-
Causal Inference | Cross-modal fusion | DPO | Data Modeling | Fraud DetectionEntry-level Full TimeSan Jose, California, United States15h ago
-
Machine Learning Engineer Intern (Monetization Technology - Ads Creative) - 2026 Start (BS/MS) USD 113K-250KA/B | A/B Testing | Algorithms | B testing | C++Internship | Social events | WorkshopsEntry-level InternshipSan Jose, California, United States15h ago
-
(General Hire) Machine Learning Engineer Intern (TikTok-Recommendation) - 2026 Summer (BS/MS) USD 119K-177KAlgorithms | C plus plus | C# | Data Structures | Deep learningDevelopment workshops | Internship program | Social eventsEntry-level InternshipSan Jose, California, United States15h ago
-
Machine Learning Engineer Intern (Trust and Safety - CV/NLP/Multimodal LLM) - 2026 Summer (BS/MS) USD 136K-237KChain-of-Thought | Computer Vision | Data Mining | Deep learning | Distributed TrainingEntry-level InternshipSeattle, Washington, United States15h ago
-
Machine Learning Engineer Intern (Trust and Safety - CV/NLP/Multimodal LLM) - 2026 Summer(BS/MS) USD 136K-237KComputer Vision | Data Mining | Deep learning | Distributed Training | Language ModelsCareer development workshops | Social eventsEntry-level InternshipSan Jose, California, United States15h ago
-
CV/NLP/Multimodal LLM Machine Learning Engineer Graduate (TikTok Trust and Safety) - 2026 Start (BS/MS) USD 119K-177KActive Learning | Computer Vision | Data Pipelines | Language Models | Language ProcessingGrowth opportunities | Onboarding programEntry-level Full TimeSan Jose, California, United States15h ago
-
Machine Learning Engineer Intern - Global E-Commerce Content Recommendation - 2026 Summer (BS/MS) USD 116K-177KAlgorithms | C plus plus | Data Structures | Deep learning | Diffusion ModelsEntry-level InternshipSeattle, Washington, United States15h ago
-
Charm Security-Forward Deployed AI Engineer USD 140K-200KAI Agents | API | Agentic Workflows | Automation | Backend DevelopmentCustomer Site Workshops | Travel opportunitiesMid-level Full TimeRemote, OR, US R16h ago
-
Member of Technical Staff - Simulation Engineer USD 180K-300KBenchmarking | C++ | Calibration | Data Generation | Distributed SystemsOn-site work in San FranciscoSenior-level Full TimeSan Francisco Bay Area22h ago
-
Mid-level Full TimeUnited States22h ago
-
AI Engineering Intern USD 60K-94KAI orchestration | APIs | Agent Frameworks | Agentic AI | CI/CDHands-on learning | Mentorship | Portfolio Ready ProjectsEntry-level InternshipFrisco, TX1d ago
-
Senior AI/ML Engineer - Agentic AI USD 148K-306KAPI Design | AWQ | Agent systems | Annotation Workflows | Application developmentSenior-level Full TimePalo Alto, CA, US, 943041d ago
-
Data Engineer - AI, Hybrid USD 137K-200KDBT | Data Cleansing | Data Imputation | Data Modeling | Data Quality401k matching contribution | 401k plan | Disability coverage | Hybrid work model | Life insuranceSenior-level Full TimeUSA-RESEARCH TRIANGLE PARK, United States R1d ago
-
Senior Data Engineer USD 140K-180KAPI Development | AWS | Airflow | Autogen | Azure401k plan with matching company contributionsSenior-level Full TimeUSA-CORP Westmont IL-Westmont - 700 Oakmont, …1d ago