Staff Machine Learning Engineer, Offline Infrastructure
Tasks
- Design and operate large scale data pipelines for training datasets
- Develop infrastructure for distributed training workflows
- Enable large scale experimentation and model iteration
- Improve reproducibility and observability of ML pipelines
- Integrate ML pipelines with workflow orchestration
- Lead architectural improvements for scalable reliable cost efficient offline ML pipelines
- Optimize performance and resource utilization in distributed compute
Perks/Benefits
- Commute subsidy
- Employee assistance program
- Employee resource groups
- Employee stock ownership
- Family leave support
- Generous vacation
- Mental health programs
- Office Food Snacks
- Relocation support not available
- Retirement pension plans
- Training and development programs
- Volunteering and donation matching program
Skills/Tech-stack
Data Lake | Data Warehouse | Data pipeline | Distributed Computing | Feature Engineering | Flink | Machine Learning | PyTorch | Python | Ray | Ray Data | Ray Train | Spark | Streaming Data | Workflow Orchestration
Education
N/A
Regions
Countries
States
Related jobs
-
API Gateway | AWS Glue | Access Control | Apache Airflow | Apache AtlasSenior-level Full TimeFAIRFAX, VA, United States3h ago
-
AI Developer - Model Creation & Full Stack USD 150K-175KAWS | Angular | Azure | CI/CD | D3.jsRemote work | USPS Public Trust Clearance eligibleMid-level Full TimeWork from home, VA, United States R3h ago
-
Enterprise Data Engineering & Operations Lead SME USD 107K-190KAWS GovCloud | Access Control | Apache Kafka | Apache Spark | Azure GovernmentSenior-level Full TimeFAIRFAX, VA, United States3h ago
-
Software/Embedded Systems Engineer USD 135K-158KAgile Development | C++ | Change Control | DoD Systems | Embedded SoftwareTravel 25%Senior-level Full TimeArlington, VA, United States3h ago
-
Data/ML Scientist SME USD 105K-150KAWS GovCloud | Anomaly Detection | Apache Spark | Bayesian Causal Inference | Bayesian MethodsMid-level Full TimeFAIRFAX, VA, United States3h ago
-
Lead AI Engineer USD 180K-280KCI/CD | ColBERT | Docker | Faiss | Fine Tuning401k match | Bonus | Childcare benefits | Dental insurance | Disability insuranceSenior-level Full TimeQuincy, MA, United States4h ago
-
API Integration | AWS | AWS Glue | Batch Processing | Code reviewSenior-level Full TimeIndianapolis, IN, United States R4h ago
-
Senior-level Full TimeUnited States5h ago
-
Senior-level Full TimeUnited States5h ago
-
Computational Biologist - Protein Engineering USD 150K-250KAWS | Amazon Web Services | CUDA | Conda | Deep learningRelocation supportEntry-level Full TimeSan Francisco, CA, US5h ago
-
Entry-level Full TimeSan Francisco, CA, US5h ago
-
Applied Math Libraries Engineer (SPU) USD 120K-170KArithmetic intensity | Atomics | BLAS | Bandwidth) | C#Entry-level Full TimeRedwood City5h ago
-
Kernel Engineer — Scientific Computing (SPU) USD 120K-170KAtomic operations | C++ | CUDA | Concurrency | DebuggingEntry-level Full TimeRedwood City5h ago
-
Senior Google Cloud Data Engineer USD 170K-240KAlerting | Anomaly Detection | Apache Beam | Apache PubSub | Batch Data ProcessingCareer advancement | Dental insurance | Health savings account | Hybrid work options | Medical insuranceSenior-level Full TimeNew York City7h ago
-
API Testing | Bash | C# | CI/CD | JUnitEntry-level Full TimePennsylvania, Exton7h ago
-
Machine Learning Engineer Intern (E-Commerce Knowledge Graph - CV/Multimodal/NLP) - 2026 Summer (BS/MS) USD 122K-243KEntity Linking | Graph Modeling | Knowledge graphs | Language Processing | Machine LearningDevelopment workshops | Social eventsEntry-level InternshipSeattle, Washington, United States8h ago
-
Senior Software Engineer, AI Coding Tools USD 244K-588KDeep learning | Fine Tuning | GPU | Inference acceleration | Language ModelsSenior-level Full TimeSan Jose, California, United States8h ago
-
Machine Learning Engineer Intern (E-commerce-Conversational AI) - 2026 Summer/Fall (PhD) USD 129K-246KChatbot Development | Information Retrieval | Language Models | Language Processing | Large Language ModelsEntry-level InternshipSeattle, Washington, United States8h ago
-
Machine Learning Engineer Intern (E-commerce-Supply Chain & Logistics) - 2026 Summer (MS) USD 136K-237KClassification | Data Mining | Deep learning | Forecasting | Knowledge graphsEntry-level InternshipSeattle, Washington, United States8h ago
-
Software Engineer, Generative AI, Workspace USD 147K-211KC++ | Distributed Systems | Generative AI | Information Retrieval | Integration TestingBenefits | Bonus | EquityMid-level Full TimeBoulder, CO, USA9h ago
-
Staff Software Engineer, Machine Learning, Google Chat USD 207K-300KAgentic Workflows | Caching | Cloud Spanner | Continuous Delivery | Continuous integrationSenior-level Full TimeSunnyvale, CA, USA9h ago
-
Software Engineer III, AI/ML Computer Vision, AR USD 147K-211KC++ | Computer Vision | Data Processing | Data Structures | Data Structures and AlgorithmsSenior-level Full TimeSan José, CA, USA9h ago
-
Agentic AI | C plus plus | C# | Cloud services | Data ProcessingMid-level Full TimeSan Francisco, CA, USA9h ago
-
Forward Deployed Engineer I, GenAI, Google Cloud USD 102K-145KAPIs | Agent systems | Cloud Computing | Cloud platform | Fine TuningSenior-level Full TimeSan Francisco, CA, USA; Calgary, AB, …9h ago
-
Senior Staff Physical AI Data Algorithm Engineer USD 203K-344KAI Agent | Artificial Intelligence | Asset Management | Cloud Computing | Data AnnotationComputational resources | Cross-team collaboration | Meals | Professional growth opportunities | SnacksSenior-level Full TimeSanta Clara, CA13h ago