Staff Software Engineer - AI Platform
Mountain View, CA, United States
USD 175K-287K Senior-level Full Time
Tasks
- Build distributed training infrastructure
- Collaborate with open-source community
- Design distributed model serving systems
- Improve system observability
- Lead AI infrastructure initiatives
- Mentor engineers
- Optimize deep learning training performance
Perks/Benefits
Skills/Tech-stack
Apache Beam | Apache Flink | Apache Spark | C# | C++ | CUDA | CUDNN | Containerization | DeepSpeed | Distributed Systems | Feature Store | GPU Inference | Go | Horovod | HuggingFace Transformers | JAX | Java | Kubeflow | Kubernetes | MLOps | MLflow | Model Parallelism | NCCL | PyTorch | Python | RDMA | Ray | Rust | Scala | Tensor Parallelism | Tensor libraries | TensorFlow | VLLM
Education
Roles
Regions
Countries
States
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 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 States8h ago
-
Principal AI Engineer - Aerospace & Defense USD 150K-180KAgentic architecture | Artificial Intelligence | Automated testing | CI/CD | CRM Integration401k company contribution | Community involvement | Disability benefits | Flexible paid time off | Flexible spending accountsSenior-level Full TimeItasca, Illinois, United States11h 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 States12h 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 States13h 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 States15h ago
-
Airflow | Apache Spark | CI/CD | Data Governance | Data LakehouseSenior-level ContractAustin, United States15h ago
-
Container Runtime | Kubernetes | Microservices | Observability | Performance TuningSenior-level Full TimeSan Jose, California, United States16h 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 States16h 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
-
Senior Software Engineer II, Storage (File & Block) USD 165K-242KBlock Storage | CSI | ClickHouse | Data durability | Distributed Storage401k match | Flexible PTO | Flexible spending account | Health savings account | Medical, dental, and vision insuranceSenior-level Full TimeSunnyvale, California1d ago
-
Senior Software Engineer II, Storage (File & Block) USD 165K-242KBlock Storage | CRD | CSI | ClickHouse | Control Plane401k employer match | Catered lunch | Childcare support | Flexible PTO | Flexible spending accountSenior-level Full TimeNew York, NY1d ago
-
Staff Software Engineer, Storage USD 188K-275KC++ | CSI | Ceph | Crash Consistency | Distributed Systems401k match | Dental insurance | Flexible PTO | Health insurance | Life insuranceSenior-level Full TimeNew York, NY1d ago
-
Staff Software Engineer, Storage USD 188K-275KC++ | CRDs | CSI | Ceph | Crash Consistency401k employer match | Employee stock purchase program ESPP | Flexible PTO | Flexible spending account | Health savings accountSenior-level Full TimeSunnyvale, California1d ago
-
Staff Software Engineer, Storage USD 188K-275KBlock Storage | C++ | CSI | Caching | Ceph401k employer match | Casual work environment | Catered lunch | Employee stock purchase program ESPP | Flexible PTOSenior-level Full TimeBellevue, Washington1d ago
-
Senior Software Engineer, Storage (File & Block) USD 153K-224KC++ | CSI | ClickHouse | Database | Debugging401k match | Childcare support | Dental insurance | Disability insurance | Employee stock purchase programSenior-level Full TimeNewYork, NY1d ago
-
Senior Software Engineer, Storage (File & Block) USD 153K-224KC++ | CSI | ClickHouse | CockroachDB | Consistency401k employer match | Catered lunch | Childcare support | Company paid life insurance | Employee stock purchase programSenior-level Full TimeBellevue, Washington1d ago
-
Senior Software Engineer, Storage (File & Block) USD 153K-224KBlock Storage | C++ | CSI | ClickHouse | Distributed Systems401k match | Disability insurance | Flexible PTO | Health insurance | Life insuranceSenior-level Full TimeSunnyvale, California1d 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