Staff Software Engineer, AI/ML Data Processing
Tasks
- Build features and Application Programming Interfaces
- Collaborate on ML ecosystem integrations
- Drive development of high performance data processing
- Enable online data processing for large scale training
- Integrate training checkpointing and inferencing components
- Optimize performance and efficiency
- Redefine data pipelines with JAX and ML Pathways
- Work with first party users on data pipeline modernization
Perks/Benefits
- N/A
Skills/Tech-stack
APIs | Checkpointing | Cloud technologies | Data Processing | Debugging | Distributed Systems | Fine Tuning | Inference | Information Retrieval | JAX | Language Processing | Machine Learning | Model Deployment | Model Evaluation | Natural Language | Natural Language Processing | Online Data Processing | Online data | Reinforcement Learning | Speech Recognition
Education
Roles
Regions
Countries
States
Cities
Related jobs
-
Research Engineer, FAIR AI & Society USD 117K-173KAI alignment | Apache Hive | Apache Spark | Artificial Intelligence | DPOEntry-level Full TimeMenlo Park, CA | New York, …2h ago
-
AI Research Engineer, FAIR Chemistry USD 147K-208KApplied Mathematics | Artificial Intelligence | Computational modeling | Computational statistics | Data ScienceCollaborative research environment | Open source contributions | Reproducible researchSenior-level Full TimeSan Francisco, CA2h ago
-
AI Compression | AV1 | AV2 | Audio CODEC | Automated testingSenior-level Full TimeBellevue, WA | Menlo Park, CA …2h ago
-
Software Engineer, Video AI/ML Specialist USD 141K-208KAV synchronization | AV1 | AV2 | Audio CODEC | Automated testingMid-level Full TimeBellevue, WA | Menlo Park, CA2h ago
-
Software Engineer, Systems USD 221K-240KAlgorithms | CSS | Data Analysis | Data Modeling | Data ProcessingEntry-level Full TimeBellevue, WA2h ago
-
C++ | Debugging | Distributed Systems | Google Cloud | InfrastructureSenior-level Full TimeMadison, WI, USA2h ago
-
AI infrastructure | Accelerator Virtualization | Container Runtime | Distributed Systems | GPUSenior-level Full TimeSeattle, WA, USA; Kirkland, WA, USA2h ago
-
Staff Software Engineer, AI/ML, YouTube USD 207K-300KAudio Processing | Data Processing | Debugging | Distributed Systems | Fine TuningSenior-level Full TimeSan Bruno, CA, USA2h ago
-
Senior Staff Software Engineer, AI/ML, Google Cloud USD 262K-365KData Processing | Data Structures | Data Structures and Algorithms | Debugging | Fine TuningSenior-level Full TimeSunnyvale, CA, USA2h ago
-
Senior Software Engineer, AI/ML, Google Cloud AI USD 174K-252KC++ | Data Processing | Data Structures | Data structures algorithms | DebuggingSenior-level Full TimeMountain View, CA, USA2h ago
-
API Development | Accessible technology | Algorithms | Code Quality | Code reviewSenior-level Full TimeSunnyvale, CA, USA2h ago
-
Algorithms | C++ | Data Structures | Data Visualization | DebuggingSenior-level Full TimeSunnyvale, CA, USA2h ago
-
Senior Software Engineer, AI and Infrastructure USD 174K-252KArtificial Intelligence | C++ | Code review | Data Storage | Data StructuresSenior-level Full TimeSunnyvale, CA, USA2h ago
-
2026 Summer Internship | Data Analytics & AI USD 50K-90KDashboarding | Data Engineering | Data Pipelines | Data analytics | ETLCareer development opportunities | Community service opportunities | Executive spotlights | Hybrid schedule | Professional development sessionsEntry-level InternshipMiramar, FL, US, 331326h ago
-
Senior Machine Learning Engineer USD 150K-200KDistributed Systems | Feature Engineering | Feature Selection | Language Models | Language Processing401k matching | Cell phone and internet stipend | Employee stock purchase plan | Flexible time off | Learning programsSenior-level Full TimeRemote - USA R9h ago
-
Senior-level Full TimeSan Jose, CA10h ago
-
Research Engineer, Gemini AutoRater USD 166K-244KData collection | Fine Tuning | Foundation Models | Human evaluation | Language ModelsSenior-level Full TimeMountain View, California, US11h ago
-
Software Engineer - Reference Data (Java/SQL) USD 175K-200KAPI Development | Distributed Systems | Git | HBase | HadoopMid-level Full TimeNew York11h ago
-
Senior LLM Software Engineer USD 191K-287KAWS | Agentic Automation | Data Pipelines | Event Driven | Event-driven architecture401k retirement plan | Caregiver leave | Commuter benefits | Dental insurance | Disability insuranceSenior-level Full TimeCosta Mesa, California, United States11h ago
-
Senior-level Full TimeHybrid (Salt Lake City, UT, US) R12h ago
-
Machine Learning Engineer, Geometry Team USD 175K-215K3D Spatial Reasoning | C++ | Computer Vision | Distributed Training | JAXSenior-level Full TimeKirkland, Washington, United States13h 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 States14h ago
-
Senior-level Full TimeRedmond, WA, US15h ago
-
Senior Staff AI Engineer, AI Privacy Expert USD 198K-326KAI privacy | Data Memorization | Differential Privacy | Explainability | Federated LearningHealth and wellness programs | Time offSenior-level Full TimeMountain View, CA, United States15h ago
-
Sr. Staff AI Engineer, GenAI Safety USD 191K-315KAI Safety | Content Safety | Distillation | Distributed Systems | Evaluation PipelineHealth and wellness programs | Time offSenior-level Full TimeMountain View, CA, United States15h ago