Sr. Staff Software Engineer, AI Infra
Mountain View, CA, United States
USD 198K-326K Senior-level Full Time
Tasks
- Collaborate with open-source community
- Design implement and optimize distributed model training and serving
- Improve observability and system understandability
- Lead concurrent AI infrastructure initiatives
- Mentor engineers and define technical culture
- Own technical strategy for AI infrastructure
Perks/Benefits
- N/A
Skills/Tech-stack
C++ | CUDA | DeepSpeed | Distributed Training | GNN | GPU Inference | Go | Horovod | Hugging Face | Hugging Face Transformers | Information Retrieval | JAX | Java | Kubeflow | LLM serving | MLflow | Model Parallelism | PyTorch | Python | Ray | Rust | Scala | Tensor Parallelism | TensorFlow | VLLM
Education
Regions
Countries
States
Related jobs
-
Software Engineer - Robot Link Platform USD 160K-232KAPI Design | Apache | Caching | Cloud services | DatabasesMid-level Full TimeFoster City, CA4h ago
-
Software Engineer - Reference Data (Java/SQL) USD 175K-200KAPI Development | Distributed Systems | Git | HBase | HadoopMid-level Full TimeNew York6h ago
-
AI Search | AWS | Amazon SageMaker | Azure AI | Azure AI Search401k | Dental insurance | Medical insurance | Paid sick hours | Vision insuranceSenior-level Contract Full TimeRidgefield Park, NJ, United States8h ago
-
AWS S3 | Ansible | Automation | Azure Blob | Azure Blob Storage401k | Dental insurance | Medical insurance | Paid sick hours | Vision insuranceSenior-level Contract Full TimeRidgefield Park, NJ, United States8h ago
-
Senior-level Full TimeSan Jose, California, United States11h ago
-
Principal Staff Software Engineer, AI Training Platform USD 207K-340KAutomatic mixed precision | C++ | Container Orchestration | Containerization | Data parallelismCommute to office based on team needs | Hybrid work options | Mentorship and career growth | Work from homeExecutive-level Full TimeMountain View, CA, United States11h ago
-
Staff Analytics Engineer - Foundational Data USD 194K-240KAirflow | Apache Spark | Business Intelligence | Data Modeling | Data ProcessingEmployee travel credits | Remote eligibleSenior-level Full TimeUnited States11h ago
-
Senior Specialist AI Engineer - Foundations USD 139K-229KCUDA | Deep learning | Distributed Training | GPU Acceleration | Generative AIHealth and wellness programs | Time awaySenior-level Full TimeSunnyvale, CA, United States11h ago
-
Senior Data Engineer USD 146K-180KAWS | AWS CDK | Amazon Redshift | Apache Airflow | CircleCI401k matching | Disability insurance | Flexible PTO | Free therapy access for employees and one household member | Friends and family discountSenior-level Full TimeNew York, NY (Hybrid) R11h ago
-
AWS | Azure | Data Modeling | Data Warehousing | Data integrationRelocation opportunitiesSenior-level Contract Full TimeSan Jose, California, United States11h ago
-
Senior AI Engineer USD 139K-229KAnt | Configuration Management | Data Mining | Data Science | Deep learningHealth and wellness programs | Hybrid work schedule | Time offSenior-level Full TimeSunnyvale, CA, United States11h ago
-
Staff AI Engineer USD 170K-277KBig Data | Data Mining | Deep learning | Information Retrieval | JavaHealth and wellness programs | Hybrid work | Paid time offSenior-level Full TimeSunnyvale, CA, United States11h ago
-
Staff Software Engineer, Ads Data Platform USD 221K-279KApache Airflow | Apache Spark | Big Data | DBT | DatabricksAnnual equity refresh grants | Equity grants | Flexible work location | Remote workSenior-level Full TimeUnited States - Remote R11h ago
-
Quantum Architect – Simulation USD 120K-170KC++ | CI/CD | DevOps | Error correction | Error modellingSenior-level Full TimePalo Alto, California, United States11h ago
-
Full-Stack AI Engineer USD 140K-190KAWS | Authentication | Authorization | Azure | CI/CD401k | Collaborative work environment | Health, dental, and vision insurance | Paid time off | Professional development opportunitiesSenior-level Full TimeWashington, D.C.11h ago
-
Senior Data Engineer USD 143K-229KAnalysis Services Tabular | Azure Analysis | Azure Analysis Services | Azure Analysis Services Tabular | Azure DataMentorship | Remote work | Team collaborationSenior-level Full TimeKansas City, MO, United States R13h ago
-
Senior Data Engineer - Data Science USD 125K-206KApache Hive | Apache Spark | D3.js | Data Engineering | Data GovernanceHealth and wellness programs | Hybrid work | Time awaySenior-level Full TimeMountain View, CA, United States13h ago
-
Software Engineer 1 USD 144K-220KAgile | Ansible | Apache NiFi | Apache Pig | IAVADental insurance | Employee referral bonus | Health savings account | Identity theft protection | Legal protectionMid-level Full TimeLinthicum Heights, Maryland13h ago
-
Staff Backend Software Engineer USD 190K-261KCI/CD | Cloud infrastructure | Distributed Systems | GPU Orchestration | GoSenior-level Full TimeNew York14h ago
-
Machine Learning Co-Op (Fall 2026) USD 96K-176KData Analysis | Data Visualization | Machine Learning | Power BI | PythonMid-level Full TimeCanton, OH, United States15h ago
-
Machine Learning Co-Op (Summer 2026) USD 96K-176KMachine Learning | Power BI | Python | SQL | StatisticsMid-level Full TimeCanton, OH, United States15h ago
-
Senior Embedded Software Engineer USD 103K-152KAsyncio | C# | CUDA | Computer Vision | Debugging401k | Commuter benefits | Company events | Flexible PTO | Health and wellness stipendSenior-level Full TimeSan Francisco, CA15h ago
-
Machine Learning Engineer - Sr. Consultant Level USD 152K-243KAgile | Big Data | C++ | CI/CD | Cloud Computing401 K | Dental insurance | FSA/HSA | Health insurance | Life insuranceSenior-level Full TimeAustin, TX, United States15h ago
-
AWS | Agile | Bash | CI/CD | Cloud ComputingHealth insurance | Paid leave | RetirementSenior-level Full TimeFort Belvoir, VA16h ago
-
Junior Data Engineer USD 79K-110KData Analysis | Data Pipeline Development | Data Transformation | Data pipeline | Pipeline DevelopmentCareer growth opportunities | Flexible work optionsEntry-level Full TimeFAIRFAX, VA, United States16h ago