Staff Software Engineer, ML Training and Inference Infrastructure
630 - 660 Hansen (Palo Alto California)
USD 228K-285K Senior-level Full Time
Tasks
- Accelerate transformer model training and inference
- Design train and deploy large deep learning models
- Optimize deep learning training on NVIDIA GPUs at scale
- Optimize model inference latency
- Optimize model pre processing and post processing
Perks/Benefits
Skills/Tech-stack
CUDA | Distributed Training | NCCL | NVIDIA GPU | NVIDIA TensorRT | Profiling | PyTorch | PyTorch Lightning | Ray | Transformers | Triton
Education
Roles
Regions
Countries
States
Cities
Related jobs
-
AC DC Sizing | AC/DC | Arc Flash | Artificial Intelligence | AutomationMultiple project extensions | Paid on time | Remote work optionSenior-level Full TimeJuno Beach, FL R22h ago
-
C++ | Computer Vision | Data Processing | Deep learning | Domain AdaptationEntry-level Full TimeSan Jose, California, United States1d ago
-
Benchmarking | C plus plus | CUDA | Computational Engineering | Distributed Computing401k match | Generous paid time off | Medical, dental & vision coverage | Stock options | WFH Hardware ReimbursementSenior-level Full TimeSunnyvale, CA R1d ago
-
Apple Neural Engine | C++ | CPU | Compilers | Computer ArchitectureMid-level Full TimeSeattle1d ago
-
On-Device ML Infrastructure Engineer (CoreML Runtime), Graphics, Games and Machine Learning USD 170K-296KC++ | Core ML | GPU Programming | JAX | LLVMMid-level Full TimeCupertino1d ago
-
Apple Neural Engine | C plus plus | CPU performance | Compilers | Computer ArchitectureMid-level Full TimeCupertino1d ago
-
AI Developer - Model Creation & Full Stack USD 130K-145KAWS | Aiohttp | Angular | Asyncio | Azure DatabricksRemote work | USPS Public Trust Clearance supportMid-level Full TimeWork from home, VA, United States1d ago
-
Senior Machine Learning Engineer USD 160K-200KAWS | CI/CD | Computer Vision | Data Processing | DockerCommuter benefits | Dental insurance | Employee assistance program | Equity | Health insuranceSenior-level Full TimeArlington, VA R1d ago
-
Senior Computer Vision & Machine Learning Engineer USD 140K-206KAblation Studies | CI/CD | Data Pipelines | Docker | Error AnalysisAutonomy | Client project impact | Support for professional growthSenior-level Full TimeRemote, US R1d ago
-
Computer Vision & Machine Learning Engineer USD 150K-200KAblation Studies | CI/CD | Computer Vision | Data Pipelines | DockerEntry-level Full TimeRemote, US R1d ago
-
C# | C++ | Fine Tuning | Foundation | ICUMid-level Full TimeCupertino1d ago
-
Staff Tech Lead, Machine Learning Engineer, Perception USD 251K-310KC plus plus | Computer Vision | Data pipeline | JAX | Machine LearningAnnual bonus | Company benefits program | Equity incentive planSenior-level Full TimeMountain View, California, United States; San …1d ago
-
HomeKit Machine Learning Engineer USD 166K-284KC# | C++ | Computer Vision | Data Pipelines | Deep learningEntry-level Full TimeSan Francisco Bay Area1d ago
-
Senior Software Engineer - Frontend USD 130K-300KAI coding | AI coding tools | Algorithms | Coding Tools | Data StructuresCareer development opportunities | Community guilds | Employee stock purchase plan | Free mental health benefit | Hybrid work environmentSenior-level Full TimeNew York, New York, USA R1d ago
-
Apache Spark | CI/CD | Deep learning | Fine Tuning | GoSenior-level Full TimeSeattle1d ago
-
AI/ML Performance Analysis and Validation Engineer USD 170K-300KComputer Architecture | CoreML | Data center | Data center architecture | Distributed ComputingMid-level Full TimeCupertino1d ago
-
Machine Learning Intern USD 100K-159KAblation Studies | Audio Processing | Deep learning | GPU Computing | Generative ModelingAccess to tools | Mentorship | Return OfferEntry-level InternshipSan Francisco1d ago
-
AI Solutions Engineer USD 138K-155KC# | C++ | Calculus | Clustering | Convolutional Neural NetworksEntry-level Full TimeDallas, TX, United States1d ago
-
AWS | Agent systems | Apache Spark | Azure | Cloud platformOn site customer collaboration | Remote work | Travel for customer visitsSenior-level Full TimeWashington, D.C. R1d ago
-
AWS | Agent systems | Apache Spark | Apache Spark SQL | AzureConference presentations | Periodic onsite work | Remote work possible | Travel every 4 to 8 weeksSenior-level Full TimeVirginia R1d ago
-
Cloud deployment | Data Processing | Docker | Feature Engineering | Model ValidationRemote workMid-level ContractUnited States (Chicago)2d ago
-
Research Engineer, Robot Learning USD 125K-175KComputer Vision | Data Pipelines | Debugging | Imitation Learning | JAXFull-time role | On-site work | Work directly with foundersMid-level Full TimeSan Francisco, CA, US2d ago
-
Mid-level Full TimeSan Francisco Bay Area2d ago
-
AI Security Engineer USD 150K-160KAPI Gateway | AWS | Adversarial Testing | Azure | CI/CD401k retirement plan | Employee assistance program | Health savings account | Life insurance | Medical, dental, and visionMid-level ContractBolingbrook, Illinois, USA R2d ago
-
Senior Software Engineer, Machine Learning (LLM & Agentic AI - TikTok - Content Ecology Local Services) USD 194K-355KCausal Inference | Causal Uplift | Causal Uplift Modeling | Computer Vision | Conversion modelingSenior-level Full TimeSan Jose, California, United States2d ago