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 engine
- Build compilation optimization system for recommendation machine learning engine
- Design graph operator fusion and automatic operator generation
- Optimize compilation strategies using hardware characteristics
- Optimize model import conversion and code generation
- Perform hardware/software co-design
Perks/Benefits
- N/A
Skills/Tech-stack
Automatic Operator Generation | Code generation | Compilation Optimization | Graph optimization | Hardware-Software Codesign | Hardware/software | Machine Learning | Model Conversion | Operator fusion | PyTorch
Education
N/A
Related jobs
-
Deloitte Technology-US Associate GenAI Engineer USD 70K-86K.Net Core | AWS | Agile | Algorithms | AngularMid-level Full TimeHermitage, Tennessee, United States5h ago
-
Consultant - Data Science / Data Lake USD 107K-188KAmazon Web Services | Apache Hadoop | Apache Kafka | Apache Spark | Apache StormEntry-level Full TimeChicago, Illinois, United States5h ago
-
Java Full Stack Developer-Software Engineer II USD 93K-155KAPI Design | AWS | Ansible | Benchmarking | Best practicesMid-level Full TimeDallas, Texas, United States5h ago
-
Anthropic Forward Deployed Engineer - GPS USD 97K-171KAPI Integration | AWS | Airflow | Anthropic Claude | Anthropic Claude APIMentorship | Professional development opportunities | Reasonable accommodations | Travel 50% | US government security clearance supportMid-level Full TimeArlington/Rosslyn, Virginia, United States; Atlanta, Georgia, …5h ago
-
API Development | AWS | CI/CD | Distributed Systems | Docker401k match | Dental insurance | Life insurance | Medical insurance | Paid time offSenior-level Full TimeNew York5h ago
-
Ads Systems | Algorithms | Computer Vision | Language Processing | Machine LearningEntry-level InternshipSan Jose, California, United States5h ago
-
Software Engineer, SystemML - AI Networking USD 170K-251KC# | C++ | CUDA | Data-parallel | Deep learningMid-level Full TimeMenlo Park, CA6h ago
-
Code Safety | Data Processing | Debugging | Distributed Systems | Fine TuningSenior-level Full TimeMountain View, CA, USA6h ago
-
Senior Software Engineer, Generative AI, Core ML USD 174K-252KAgentic Applications | Automated testing | Computer Vision | Data Processing | DebuggingSenior-level Full TimeMountain View, CA, USA6h ago
-
Software Engineer III, Generative AI, Agent Engineering USD 147K-211KArtificial Intelligence | BigQuery | CI/CD | Cloud Storage | Computer VisionSenior-level Full TimeKirkland, WA, USA6h ago
-
Forward Deployed Engineer III, Google Cloud, Applied AI USD 174K-252KAPIs | Agent systems | Cloud platform | Conversational AI | DebuggingTravel opportunitiesSenior-level Full TimeSan Francisco, CA, USA; Atlanta, GA, …6h ago
-
Artificial Intelligence | Automated Evaluation | Behavior analytics | C# | C++Senior-level Full TimeMountain View, CA, USA6h ago
-
Artificial Intelligence | Documentation | Gemini | Hugging Face | JAXMid-level Full TimeMountain View, CA, USA6h ago
-
Forward Deployed Engineer II, Applied AI, Google Cloud USD 147K-211KAPI Integration | APIs | Agent systems | Cloud platform | Evaluation PipelinesTravel opportunitiesMid-level Full TimeSan Francisco, CA, USA; Atlanta, GA, …6h ago
-
Senior Staff Software Engineer, Generative AI USD 262K-365KComputer Vision | Data Processing | Debugging | Deep learning | Distributed ComputingSenior-level Full TimeSunnyvale, CA, USA6h ago
-
Junior AI Engineer (Open to remote) USD 110K-135KAPI Development | Language Model | Language Model Evaluation | Language Models | Language Processing401k | Dental insurance | Health savings account | Medical insurance | Paid time offEntry-level Full TimeNew York, NY, US, NY 10019 R10h ago
-
Robotics Application Engineer USD 100K-300KAutonomous Systems | C++ | Computer Vision | Machine Learning | PythonMid-level Full TimePittsburgh, San Mateo12h ago
-
Machine Learning Intern USD 80K-90KAWS | Apache Hive | Apache Spark | C++ | Cloud platformFinal evaluation | Mentorship | Performance feedbackEntry-level InternshipMountain View, USA16h ago
-
AWS Cloud | Amazon Bedrock | Amazon SageMaker | Data analytics | Deep learningSenior-level Full TimeNew York, New York, USA17h ago
-
AWS | Data Governance | Data Lineage | Data Modeling | Data PipelinesMid-level Full TimeHawthorne, California, USA17h ago
-
AI Engineer USD 120K-158KAI Services | AIOps | Azure | Azure AI | Azure AI Services401k match | Dental insurance | Holidays | Medical insurance | Paid time offMid-level Full TimeLos Angeles, CA, USA18h ago
-
Senior Machine Learning Engineer, GenAI Data USD 243K-295KAmazon S3 | Batch Processing | C plus plus | Data Pipelines | Data PreprocessingSenior-level Full TimeSan Mateo, CA, United States R18h ago
-
Software Engineer, Cortex AI Infrastructure USD 160K-225KC++ | CI/CD | Cloud Native | Cloud Native Architecture | Data-intensiveMid-level Full TimeUS-CA-Menlo Park18h ago
-
Senior Applied AI Engineer USD 200K-270KA/B | A/B Testing | AWS | Ablation Testing | Agentic WorkflowsCareer growth | Mentorship | Travel Onsite TimeSenior-level Full TimeUS-CA-Menlo Park19h ago
-
Senior Software Engineer, Machine Learning USD 190K-220KAWS | Airflow | DBT | Kubernetes | MLflow401k match | Medical/Dental/Vision insurance | Paid Holidays | Paid parental leave | Remote-first teamSenior-level Full TimeRemote (United States) R19h ago