Technical Lead Manager, Machine Learning Runtime & Serving
Tasks
- Architect ML runtime systems for edge and cloud
- Build distributed low latency fault tolerant backend systems
- Design profiling and benchmarking infrastructure
- Develop hardware aware performance optimizations
- Extend ML compilers and runtimes
- Guide technical vision for ML infrastructure
- Identify triage and eliminate ML stack bottlenecks
- Manage and grow engineering team
- Optimize real time latency and memory usage
- Partner with ML researchers on system workloads
- Transition ML workloads to JAX native runtime
Perks/Benefits
Skills/Tech-stack
Distributed Systems | Fault Tolerance | GPU Computing | JAX | Low Latency | Low-Latency Systems | Machine Learning | ONNX Runtime | OpenXLA | PJRT | Python | TPU | TVM | TensorRT
Education
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 R6d ago
-
Featured Feat. Associate Director, Data Labs USD 167K-167KAWS | Cloud Computing | Compute Infrastructure | Data Analysis | LLM GovernanceConference speaking opportunities | Hybrid work schedule | Media appearancesSenior-level Full TimeWashington, District of Columbia, 20004, United … R19d ago
-
Sr AI Engineer USD 115K-142KAgent systems | Agentic Workflows | CI/CD | Citation Grounding | Cloud ComputingSenior-level Full TimeAda, MI, United States, 493559h ago
-
PIIM Analytics Engineer USD 86K-132KAPIs | Access Control | Azure Databricks | Data Architecture | Data GovernanceDental insurance | Medical insurance | Optional Disability Insurance | Paid time off | Supplemental insuranceMid-level Full TimeClay, NY, US11h ago
-
PIIM Analytics Engineer USD 105K-160KAI/ML | APIs | Access Control | Azure Databricks | Code AutomationFull time office or project telework | Relocation assistanceMid-level Full TimeHouston, TX, US11h ago
-
Senior-level Full TimeSeattle, Washington, United States; USA - …12h ago
-
Staff Robotics Engineer, Maritime USD 254K-336KC++ | Collision Avoidance | Computer Vision | Computer Vision Algorithms | Machine LearningHealth benefits | Mentorship | Recovery BenefitsSenior-level Full TimeQuincy, Massachusetts, United States13h ago
-
Senior Robotics Engineer, Maritime USD 220K-292KC++ | Cameras | Computer Vision | Gazebo | LidarHealth insurance | Mentorship | Travel opportunitiesSenior-level Full TimeQuincy, Massachusetts, United States13h ago
-
Machine Learning Operations Engineer USD 140K-240KAWS | Alerting | Cloud GPU | Compression | Data Quality401k | Dental insurance | Health insurance | Office food and beverages | Stock option planMid-level Full TimeRWC HQ13h ago
-
Machine Learning Engineer, GPU Kernel and Runtime USD 213K-263KCUDA | Deep learning | GPU Computing | Kernel optimization | Machine LearningSenior-level Full TimeMountain View, California15h ago
-
RFML Engineer, EW USD 166K-220K5G | C++ | Channel equalization | Channel estimation | Deep learningEmployee benefits package | Travel | U.S. Secret security clearance supportMid-level Full TimeCosta Mesa, California, United States16h ago
-
Staff ML Engineer, StockStory USD 140K-160KCNN | Deep learning | Embeddings | Human-in-the-loop | Information ExtractionDaily group classes | Daily specials | Dry cleaning | Free onsite fitness center | Free shuttle transportationSenior-level Full TimeEnglewood Cliffs, NEW JERSEY, United States17h ago
-
Staff Engineer, Data Analysis (R5290) USD 121K-197KAnomaly Detection | Automated Validation | Cloud Storage | Communications | DashboardsSenior-level Full TimeUnited States17h ago
-
Batch scheduling | Batching | CI/CD | Caching | Capacity PlanningFlexible working hours | Generous vacation | Optional equity donation matching | Parental leaveSenior-level Full TimeSan Francisco, CA17h ago
-
Staff Research Engineer, Discovery Team USD 350K-850KArtificial Intelligence | Cloud deployment | Containerization | Data Pipelines | Distributed SystemsFlexible remote work policy | Flexible working hours | Generous vacation | Parental leave | Visa sponsorship assistanceSenior-level Full TimeSan Francisco, CA17h ago
-
Software Engineer, Systems USD 150K-250KAWS S3 | Apache Iceberg | Apache Parquet | C# | C++401k match | Catered meals | Commuter benefit | Dental insurance | Flexible PTOMid-level Full TimeSan Francisco17h ago
-
Senior AI Solutions Engineer USD 164K-215KAI Observability | Agentic Systems | Airflow | Amazon S3 | Amazon Web Services401k plan | Annual Caltrain Pass | Fastrak reimbursement | Fitness reimbursement | Health, dental, vision insuranceSenior-level Full TimeUS - Remote R18h ago
-
C++ | Deep learning | Distributed Training | ETL Pipelines | GoSenior-level Full TimeMountain View, CALIFORNIA, United States18h ago
-
C++ | Deep learning | Distributed Training | ETL | GoSenior-level Full TimeMountain View, CALIFORNIA, United States18h ago
-
Senior-level Full TimeOakland, CA, United States18h ago
-
Test Data Engineer USD 68K-131KArtificial Intelligence | Automation | Cloud | Configuration Management | Data Governance401k match | Catered lunches | Coffee tea beverages | Dental insurance | Health insuranceMid-level Full TimeBoston19h ago
-
Active Learning | Agentic Workflows | Bayesian optimization | Computational Biology | Experimental DesignSenior-level Full TimeSomerville, MA19h ago
-
AI Data Engineer USD 133K-185KAmazon Redshift | BigQuery | CI/CD | Claude Code | Clay401(k) self contribution | BetterUp coaching | Company breaks | Development plan | Flexible paid time offSenior-level Full TimeAustin, TX19h ago
-
Data Visualization Engineer USD 90K-112KArcGIS | Cartography | D3 | Data Visualization | GDALCommuter benefits | Employee assistance program | Health savings account | Home office reimbursement | LinkedIn Learning accessMid-level Full TimeSan Francisco, CA R19h ago
-
Senior Machine Learning Engineer, Agent Oversight USD 216K-270KAgent Orchestration | Anomaly Detection | Drift Detection | Evaluation Methodologies | ExperimentationDental insurance | Health insurance | Learning and development stipend | Paid time off | Retirement benefitsSenior-level Full TimeSan Francisco, CA; New York, NY19h ago