MLOps Engineer (JAX, PyTorch, Pallas/Triton)
Tasks
- Create evaluation rubrics and scoring frameworks
- Design MLOps and machine learning systems tasks
- Develop solutions for ML infrastructure problems
- Improve reasoning capabilities of advanced AI systems
- Maintain consistency and technical accuracy across training datasets
- Optimize distributed training architectures
- Partner with research teams to improve model capabilities
- Review and evaluate technical solutions
- Tune infrastructure performance
Perks/Benefits
Skills/Tech-stack
Distributed Training | Evaluation Methodologies | GPU Kernel | GPU kernel programming | Infrastructure Optimization | JAX | Kernel programming | ML Pipeline | MLOps | Machine Learning | Machine Learning Infrastructure | Pallas | Performance Tuning | PyTorch | Triton
Education
N/A
Roles
Related jobs
-
Software Engineer 2 - (Python, LLM, Machine Learning) USD 201K-246KAPI Integration | AWS | Agentic Workflows | Azure | CrewAI401k match | Educational assistance | Federal Holidays | Healthcare coverage | Paid time offMid-level Full TimeAnnapolis Junction, Maryland R9h ago
-
Staff AI Engineer - Zoom AI Services Platform USD 177K-387KAI Inference | AI Platform | AI Platform Engineering | API Design | C++Hybrid work | Remote workSenior-level Full TimeSeattle (WA), United States R1d ago
-
Data Engineer, Mid USD 77K-176KAgentic AI | Apache Spark | Artificial Intelligence | Data Mining | Data ModelingDependent care | Disability insurance | Health insurance | Life insurance | Paid leaveMid-level Full TimeUSA, CO, Aurora (3855 Lewiston St), … R1d ago
-
Staff Software Engineer - Agentic AI Platform USD 165K-209KAWS | Agent systems | Agile | Apache Spark | C#Career development | Employee resource groups | Flexible WFH policy | Generous PTO policy | Paid volunteer timeSenior-level Full TimeUS-Texas-Austin, United States R1d ago
-
Machine Learning Engineer USD 238K-345KAPI Development | AWS | Agent Orchestration | Azure | Cloud platformSenior-level Full TimeSan Jose, United States R1d ago
-
Analytics and AI Developer USD 73K-102KClassification | Docker | Forecasting | Git | GitHubAnnual incentive plan | Full-time schedule | Hybrid work | Less Than 15 Percent TravelSenior-level Full TimeAlpharetta, United States R1d ago
-
Senior Machine Learning Engineer USD 150K-241KC++ | Cloud Architecture | Computer Vision | Data Pipelines | Fine TuningDiscretionary paid time off | Emotional and mental wellness support | Employer 401k match | Fitness programs | Learning and development programsSenior-level Full TimeSeattle, Washington, United States R1d ago
-
Machine Learning Engineer USD 212K-318KApache Spark | Data Pipelines | Deep learning | ML Systems | Machine Learning401k | Dental insurance | Medical insurance | Telecommuting | Vision insuranceMid-level Full TimeSouth San Francisco, CA R1d ago
-
Senior Staff Machine Learning Engineer - Agentic AI USD 201K-352KAPI Development | Agent Orchestration | Agent systems | C++ | Compliance401(k) plan match | ESPP | Family leave programs | Flexible spending accounts | Flexible time awaySenior-level Full TimeSanta Clara, CALIFORNIA, United States R1d ago
-
Bayesian statistics | Cloud deployment | Control Systems | Data Pipelines | Decision MakingContract-to-hire | Remote workSenior-level Contract Full TimePalo Alto, CA, United States R1d ago
-
Senior AI Engineer, Unstructured AI USD 204K-255KAPI Design | Apache Spark | Artificial Intelligence | Context engineering | Data CatalogingFlex Fund monthly stipend | Health coverage | Hybrid work | Pension/401K | Time offSenior-level Full TimeNew York, New York, USA R1d ago
-
Software Engineer, Research Data Platform USD 320K-405KAPI Design | BigQuery | Data Cataloging | Data Lineage | Data PipelinesFlexible working hours | Generous vacation | Hybrid work | Optional equity donation matching | Parental leaveMid-level Full TimeSan Francisco, CA | New York … R1d ago
-
Senior Machine Learning Engineer, Infrastructure USD 212K-318KData Validation | Distributed Systems | Feature Engineering | Feature Store | Live Inference401(k) plan matching | Commuter benefits | Company holidays | Flexible time off | HealthcareSenior-level Full TimeNew York R1d ago
-
Senior Consultant – Semantic Data & AI Engineering USD 146K-183KAPI Development | AWS | Agile | Amazon Neptune | Automated testing401k match | Commuter benefits | Dependent care | FSA | HSASenior-level Full TimeUS - New York R1d ago
-
Principal Consultant – Semantic Data & AI Engineering USD 146K-183KAPI Development | AWS | Agile | Automated testing | Azure401k match | Commuter benefits | Dependent care | FSA | HSASenior-level Full TimeUS - New York R1d ago
-
Software Engineer - Applied AI USD 170K-225KCI/CD | Context engineering | Evaluation | Go | GraphQL401k | Commuter stipend | Educational stipend | Health insurance | NYC Office AccessMid-level Full TimeRemote/Hybrid R1d ago
-
Security Operations AI Engineer, Contract USD 151K-200KAI Governance | AI RMF | AI Security | Adversarial Attacks | Adversarial Machine LearningMid-level Full TimeRemote, United States R1d ago
-
Senior AI/ML Flex Solution Engineer USD 165K-216KArtificial Intelligence | Data Engineering | Data Preparation | Deployment | MLOpsSenior-level Full TimeUS-NY-Remote R1d ago
-
Computational Designer- Structural USD 106K-126KAdobe Creative | Adobe Creative Suite | BIM | C# | Computational DesignHome office stipend | Hybrid work schedule | Training opportunities | Travel to office periodicallyMid-level Full TimeKansas City, MO, United States R1d ago
-
Artificial Intelligence/Machine Learning Engineer USD 96K-216KAPI | Agentic Automation | Data integration | LLM | Language ModelsMid-level Full TimeAnnapolis Junction, MD, United States R1d ago
-
Senior Data Analytics Engineer USD 145K-206KAWS | CRM | DBT | Data Model | Data Modeling401k match | Disability insurance | Employee assistance program | Flexible spending account | Health Savings Account contributionSenior-level Full TimeBRIC, United States R2d ago
-
Sr. Data Engineer USD 114K-185KAgile | Amazon Web Services | Apache Spark | Azure | Cloud platform401k retirement savings plan | CVS store discount | Disability benefits | Education assistance | Employee stock purchase planSenior-level Full TimeIrving-750 West John Carpenter, United States R2d ago
-
Data Scientist II USD 95K-142KAirflow | CI/CD | Data Governance | Data Validation | Data Visualization401k | Employee stock purchase plan | Paid time off | Parental leave | Tuition assistanceSenior-level Full TimeRemote, United States R2d ago
-
Senior Solutions Architect, Financial Services Banking USD 184K-287KApache Spark | C++ | CUDA | Dask | Deep learningSenior-level Full TimeUS, NY, Remote R2d ago
-
Engineer 3, Machine Learning-5125 USD 126K-165KAgile | Amazon Web Services | Apache Airflow | Apache Spark | CI/CDRemote work one day per weekEntry-level Full TimeDC - Washington, 1325 G ST … R2d ago