Staff Machine Learning Engineer – AI/ML Compiler
Santa Clara, California, United States of America
USD 160K-240K Senior-level Full Time
Tasks
- Analyze profile debug compilation failures
- Automate model compilation workflows
- Build ONNX compilation paths
- Build PyTorch compilation paths
- Contribute ONNXRuntime QNN execution provider
- Design compilation pipeline
- Develop developer diagnostics
- Develop graph optimization passes
- Maintain CI CD evaluation harnesses
- Partition graphs for CPU GPU NPU
- Validate compiled model artifacts
Perks/Benefits
- N/A
Skills/Tech-stack
ATen | C++ | CI/CD | FX | Graph optimization | Graph partitioning | Lowering | MLIR | ONNX | ONNX Runtime | Operator fusion | PyTorch | Python | Shape inference | TVM | Torch compile | Torch export
Education
Regions
Countries
States
Cities
Related jobs
-
(US) Principal ML System Engineer USD 139K-304KAWS | Access Control | Alerting | Audit Logging | AuthenticationSenior-level Full TimeRemote, USA R4h ago
-
Machine Learning Engineer, API Multicloud USD 295K-445KAPIs | AWS | Data Pipelines | Data Structures | Data structures algorithmsSenior-level Full TimeSan Francisco5h ago
-
Applied Probability | Artificial Intelligence | C++ | Data Engineering | Data VisualizationExposure to senior leaders | Mentorship | Rotational projects | Technical trainingEntry-level Full Time141753-NC-Three Wells Fargo Center, Charlotte, United …8h ago
-
Senior Software Engineer (Storage) - remote in the US USD 100K-170KArgoCD | CSI | Cluster API | Flux | GitOpsCompany outings and happy hours | Conferences and working groups | Hackathons and tech talks | Professional development and trainingSenior-level Full TimeRemote, USA, United States R11h ago
-
Autonomy Algorithms Software Engineer USD 115K-145KAgent systems | Agile | Algorithms | C++ | CI/CDHybrid work environment | Remote work optionMid-level Full TimeArlington, VA R11h ago
-
Autonomy Algorithms Software Engineer USD 115K-145KAgent systems | Agile | Algorithms | C++ | CI/CDFlexible work schedule | Hybrid work environment | Remote work availableMid-level Full TimeWoburn, MA R11h ago
-
Software Engineer, Embedded Agentic AI USD 170K-210KC# | C++ | Cloud Platforms | Containerization | DockerDisability benefits | Equity awards | Health insurance | Life insurance | Paid time offEntry-level Full TimeSan Jose, California R12h ago
-
Senior Embedded Software Engineer USD 210K-265KC# | C++ | CAN | Data Processing | DebuggingBonus | Catered lunch | Company holidays | Dental insurance | EV chargingSenior-level Full TimeSunnyvale, CA12h ago
-
Senior Research Engineer, LLM Training & Post-Training USD 165K-310KCUDA | DPO | DeepSpeed | Distributed Training | Experiment design401k matching | Company holidays | Comprehensive health coverage | Dental coverage | Flexible time offSenior-level Full TimeNew York, New York, United States; … R12h ago
-
Data Engineer, Knowledge Graph USD 200K-300KAmazon Neptune | Data Modeling | Data Transformation | Data integration | Graph Algorithms401k savings program | Family leave | Health care benefits | Mental wellness programs | Parental leaveSenior-level Full TimeNew York, NY13h ago
-
Computer Vision Engineer USD 220K-250KAgentic Systems | Computer Vision | Dataset curation | Deep learning | Fine TuningSenior-level Full TimeSan Bruno, California13h ago
-
Bare Metal | Beamforming | C# | C++ | Cache optimizationE Verify Support | Health insurance | Professional development | Retirement plansSenior-level Full TimeHuntington Beach, CA13h ago
-
Specialist Solutions Architect - Data Engineering & Warehousing (Digital Native Business) USD 180K-247KAWS | Anomaly Detection | Apache Spark | Cloud platform | Data ModelingContinuous learning | Mentorship | Remote work | Travel up to 30 percentSenior-level Full TimeUnited States R14h ago
-
Enterprise Applied AI Solution Specialist USD 90K-130KAPI Integration | Agent Frameworks | Conversational Design | Data Governance | Function CallingMid-level Full TimeWeston,FL,United States R14h ago
-
Embedded Software Engineer - Access Control USD 150K-300KBash | C# | C++ | Device Drivers | Distributed SystemsCommuter benefits | Expanded mental health support | Fertility benefits | Flexible PTO | Flexible spending accountMid-level Full TimeSan Mateo, CA United States14h ago
-
Senior-level Full TimeRedwood City, California, United States14h ago
-
AWS | AWS EKS | Azure | Azure AKS | CI/CD401k match | ESPP | Family leave programs | Flexible time off | Flexible-hybrid workSenior-level Full TimeSan Diego, CALIFORNIA, United States R14h ago
-
AWS | Azure | CI/CD | Cloud platform | Containers401k matching | Company matching donations | ESPP | Family leave programs | Flexible hybrid work scheduleSenior-level Full TimeSan Diego, CALIFORNIA, United States R15h ago
-
Data Engineer I - Salesforce Data 360 USD 87K-123KAPIs | Access Controls | Data Governance | Data Lineage | Data Modeling401k match | Career growth and development programs | Comprehensive onboarding | Conference participation | Employee resource groupsMid-level Full TimeMorrisville, North Carolina (NC), United States R15h ago
-
Applied AI Engineer USD 171K-257KAMD | Agentic Systems | Alignment Frameworks | CRM | ContainerizationCompany-sponsored team events | Flexible time off | Travel opportunities | Wellness resourcesSenior-level Full TimeRemote, United States R16h ago
-
Assembly Language | C# | C++ | Debugging | Engineering Tools401k | Birthday off | Company events | Dental coverage | Equipment and supplies allowanceMid-level Full TimeAnnapolis Junction, MD16h ago
-
AWS | AWS CDK | AWS CloudFormation | AWS Database | AWS Database Migration ServiceSenior-level Full TimeSt. Louis, MO16h ago
-
AWS | AWS CDK | AWS DMS | AWS Glue | AWS LambdaSenior-level Full TimeSt. Louis, MO16h ago
-
Senior Database Engineer USD 140K-160KAWS | AWS RDS | Amazon Aurora | Amazon Redshift | CI/CD401k | Employee assistance program | HSA | Life insurance | Long-term disabilitySenior-level ContractFoster City, California, USA R16h ago
-
Staff Backend Engineer, Customer Value Optimization USD 190K-220KDistributed Systems | MLOps | Machine Learning | Microservices | PythonSenior-level Full TimeNew York, NY, United States17h ago