Machine Learning Engineer - AI Compiler Optimization
San Jose, California, United States
USD 156K-387K Mid-level Full Time
Tasks
- Adapt recommendation models from PyTorch to the engine
- Build compilation optimization system for recommendation machine learning engine
- Collaborate on hardware/software co-design
- Design full stack optimization for graph operator and memory
- Implement graph operator fusion and automatic operator generation
- Optimize compilation strategies based on hardware characteristics
- Optimize model import conversion and code generation
Perks/Benefits
- N/A
Skills/Tech-stack
Co-design | Code generation | Compiler optimization | Graph optimization | Hardware-Software Co-design | Hardware/software | Machine Learning | Model Deployment | Operator fusion | PyTorch | Software Co-design
Education
N/A
Related jobs
-
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 … R18d ago
-
Agentic AI Engineer USD 101K-151KAgentic Workflows | Debugging | Embeddings | Git | GitHubFinancial benefits | Health benefits | Wellness benefitsMid-level Full TimePennsylvania, Canonsburg5h ago
-
AI | AI Inference | CPU | Containers | GPUSenior-level Full TimeSeattle, Washington, United States7h ago
-
Senior Software Engineer, Cross Platform Applications USD 212K-387KArtificial Intelligence | Automation | CI/CD | Code Analysis | Dynamic analysisSenior-level Full TimeSan Jose, California, United States7h ago
-
Cross-Functional Collaboration | Cross-functional | Data Pipelines | Full Stack | Full Stack Machine LearningSenior-level Full TimeSan Jose, California, United States7h ago
-
A/B | A/B Testing | B testing | Behavior Modeling | ExperimentationSenior-level Full TimeSan Jose, California, United States7h ago
-
Software Engineer, Sensors & Computer Vision USD 146K-208K3D Reconstruction | C++ | Camera | Clustering | Convolutional Neural NetworksMid-level Full TimeSunnyvale, CA8h ago
-
C++ | Computer Vision | Data Processing | Debugging | Distributed ComputingSenior-level Full TimeSunnyvale, CA, USA8h ago
-
Principal Engineer, AI Ecosystem USD 307K-428KCUDA | Cloud Native | Cloud Orchestration | Deep learning | Distributed SystemsSenior-level Full TimeSunnyvale, CA, USA; Kirkland, WA, USA8h ago
-
Constrained random | Constrained random verification | Control Status Registers | Coverage analysis | Formal verificationSenior-level Full TimeSan Diego, CA, USA; Mountain View, …8h ago
-
AI Engineer, Advanced Solutions Lab, Google Cloud USD 152K-222KApache Beam | Apache Hive | Artificial Intelligence | C++ | Cloud ComputingSenior-level Full TimeChicago, IL, USA; Atlanta, GA, USA8h ago
-
Senior Software Developer, Computer Vision, XR USD 174K-253KAugmented Reality | C++ | Computer Vision | Data Processing | DebuggingSenior-level Full TimeSan Jose, CA, USA; Waterloo, ON, …8h ago
-
Bi-Encoder | Cross-Encoder | Data Processing | Data Structures | Data Structures and AlgorithmsBonus | Equity | Health benefitsSenior-level Full TimeSunnyvale, CA, USA; Kirkland, WA, USA8h ago
-
Software Engineer III, AI/ML, gUP Customer Support USD 147K-211KAgile | C++ | Code review | Data Processing | DebuggingSenior-level Full TimeSunnyvale, CA, USA8h ago
-
Business Data Scientist, Applied Machine Learning, GCS USD 138K-198KA/B | A/B Testing | B testing | Causal Inference | Causal MLMid-level Full TimeMountain View, CA, USA; New York, …8h ago
-
Staff Software Engineer, ML Data Infrastructure, YouTube USD 207K-301KAlgorithms | C++ | Data Infrastructure | Data Processing | Data StructuresSenior-level Full TimeSan Bruno, CA, USA8h ago
-
Applied AI ML Lead - Generative AI and Semantic Modeling USD 177K-215KAgent-based | Agent-based AI | Context engineering | Continuous integration | Data ContractsBackup childcare | Financial coaching | Health care coverage | Mental health support | Retirement savings planSenior-level Full TimeJersey City, NJ, United States14h ago
-
AI Architect USD 190K-230KAI Foundry | AI Search | Agent workflows | Agents | AzureInclusive benefits | Wellness programsSenior-level Full TimeUnited States17h ago
-
Applied AI Lead-Vice Presient USD 177K-215KAWS | Agent coordination | Agent workflow | Agent workflow orchestration | Agentic designSenior-level Full TimeJersey City, NJ, United States18h ago
-
Batch Processing | Bias Mitigation | C++ | Data Processing | Experimentation401k employer match | Caregiving support | Family planning support | Flexible vacation | Gender-affirming careSenior-level Full TimeRemote - United States R18h ago
-
Machine Learning Engineer USD 151K-265KC++ | Cloud infrastructure | Data Analysis | Debugging | Distributed ComputingSenior-level Full TimeSan Jose, United States R19h ago
-
Data Scientist, Analytics and Modelling USD 169K-180KAWS Glue | AWS Lambda | AWS S3 | Agile | Amazon RedshiftMid-level Full TimeNew York, 745 7th Avenue, United …19h ago
-
AI/ML Engineer USD 99K-225KAWS | Azure | Confluent | Deep learning | DockerPaid leave | Professional development | Tuition assistanceMid-level Full TimeUSA, VA, Arlington (1550 Wilson Blvd), …19h ago
-
Machine Learning Research Engineer USD 140KComputer Vision | Data Assimilation | Data Engineering | Data Pipelines | Data cleaning401k | Dental insurance | Health insurance | Office food and beverages | Stock option planMid-level Full TimeRWC HQ20h ago
-
Data Analysis | Experiment design | Geospatial Data | Machine Learning | PyTorch401k | Dental insurance | Health insurance | Office food and beverages | Stock option planMid-level Full TimeRWC HQ20h ago