Machine Learning Engineer - AI Compiler Optimization
San Jose, California, United States
USD 184K-287K (estimate) Senior-level Full Time
Tasks
- Adapt recommendation models from PyTorch to engine
- Build compilation optimization system
- Conduct performance bottleneck analysis
- Design full stack optimization solutions
- Generate automatic operators
- Implement graph operator fusion
- Optimize compilation strategies for hardware characteristics
- Optimize low-latency inference
- Optimize model import conversion and code generation
- Perform hardware/software co-design
Perks/Benefits
- N/A
Skills/Tech-stack
Co-design | Code generation | GPU compilation | Hardware-Software Co-design | Hardware/software | Loop Optimization | MLIR | Machine Learning | Memory Optimization | Model Conversion | NPU compilation | Operator optimization | Performance Profiling | PyTorch | Software Co-design | TVM | TensorFlow | Triton
Education
N/A
Regions
Countries
States
Cities
Related jobs
-
Senior Research Engineer / Scientist -AI for Databases USD 207K-368KAWS | Architecture Search | Artificial Intelligence | Automated Model Tuning | C++401k match | Dental insurance | Disability coverage | Health insurance | Life insuranceSenior-level Full TimeSeattle, Washington, United States R5h ago
-
Machine Learning System Scheduling Engineer Graduate (Applied Machine Learning) - 2027 Start USD 128K-256KAlgorithms | CUDA | Cloud Storage | Computer Science | ContainerizationEntry-level Full TimeSan Jose, California, United States5h ago
-
Student Researcher (AI Foundation Models Infrastructure - Seed Infra) - 2026 Start (PhD) USD 162K-162KC++ | CUDA | Compiler optimization | Distributed Computing | Machine LearningHealth insurance | Housing allowance | Life insurance | Paid Holidays | Paid sick timeEntry-level Full TimeSeattle, Washington, United States5h ago
-
ARM Cortex | ARM Cortex M4 | AV1 | AVC | C++401-k match | Dental insurance | Life insurance | Long-term disability | Medical insuranceEntry-level Full TimeSan Diego, California, United States5h ago
-
AUC Optimization | Apache Flink | Apache Hadoop | Apache Spark | Big Data401k savings plan | Dental insurance | Life insurance | Long-term disability | Medical insuranceEntry-level Full TimeSan Jose, California, United States5h ago
-
Algorithms | CUDA | Computer Architecture | Data Structures | Deep Learning Training401k match | Life insurance | Long-term disability | Medical/Dental/Vision insurance | Paid HolidaysEntry-level Full TimeSan Jose, California, United States5h ago
-
Machine Learning Engineer USD 170K-250KAzure ML | Azure ML Endpoints | Azure Machine Learning | Bayesian Methods | Boosted TreesSenior-level Full TimeAustin, Texas | Fremont, California | …6h ago
-
Deep Neural Networks | Ensemble learning | Graph theory | Hadoop | HiveHealth insurance | Housing allowance | Life insurance | Paid Holidays | Paid sick timeEntry-level InternshipSeattle, Washington, United States6h ago
-
Data Structures | Data Structures and Algorithms | Deep learning | Ensemble Methods | Graph theory401k match | Dental insurance | Life insurance | Long-term disability | Medical insuranceEntry-level Full TimeSeattle, Washington, United States6h ago
-
Data Querying | Data Structures | Deep Neural Networks | Ensemble learning | Graph theory401k matching | Dental insurance | Life insurance | Long-term disability | Medical insuranceEntry-level Full TimeSan Jose, California, United States6h ago
-
Data Structures | Data Structures and Algorithms | Deep Neural Networks | Ensemble trees | Graph theoryHealth insurance | Housing allowance | Life insurance | Paid Holidays | Paid sick timeEntry-level InternshipSan Jose, California, United States6h ago
-
AWS | Azure | Azure ML | C++ | CUDAEntry-level Full TimeSeattle, Washington, United States6h ago
-
Research Engineer / Scientist -AI for Databases USD 153K-300KArchitecture Search | Artificial Intelligence | Automated Database Tuning | C++ | Database Internals401k match | Life insurance | Long-Term Disability coverage | Medical/Dental/Vision insurance | Paid HolidaysMid-level Full TimeSeattle, Washington, United States R6h ago
-
Machine Learning Engineer, AI Coding Tools USD 162K-387KC++ | CUDA | CUDA Deployment Optimization | Deep learning | Deployment optimization401k savings plan | Dental insurance | Life insurance | Medical insurance | Paid HolidaysMid-level Full TimeSan Jose, California, United States6h ago
-
Computer Vision | Data Processing | Deep learning | Language Models | Large Language ModelsSenior-level Full TimeSan Jose, California, United States6h ago
-
C++ | Computer Vision | Deep learning | End to End | End to End Machine Learning Pipelines401k match | Dental insurance | Life insurance | Long-term disability | Medical insuranceMid-level Full TimeSan Jose, California, United States6h ago
-
Production Engineer - Applied Machine Learning USD 162K-387KAlerting | Auto Scaling | Auto rollback | C++ | CI/CD401k match | Dental insurance | Life insurance | Long-term disability | Medical insuranceMid-level Full TimeSan Jose, California, United States6h ago
-
Alignment for Large Language Models | C++ | Deep learning | Fine Tuning | GoHealth insurance | Housing allowance | Life insurance | Paid Holidays | Paid sick timeEntry-level InternshipSan Jose, California, United States6h ago
-
3D Reconstruction | CTC | Computer Vision | Conformer | Convolutional Networks401k match | Dental insurance | Life insurance | Long-term disability | Medical insuranceSenior-level Full TimeSan Jose, California, United States6h ago
-
C++ | Container Runtime | Distributed Systems | Distributed tracing | DockerSenior-level Full TimeSeattle, Washington, United States6h ago
-
Machine Learning Engineer - Orchestration USD 162K-250KApache Flink | Apache Mesos | Auto Scaling | Celery | Deep learningMid-level Full TimeSan Jose, California, United States6h ago
-
Data Processing | Deep learning | Distributed Computing | Feature Engineering | Golang401k plan with company match | Dental insurance | Health insurance | Life insurance | Long-Term Disability coverageMid-level Full TimeSan Jose, California, United States6h ago
-
Anomaly Detection | Apache Flink | Apache Spark | C++ | ClassificationHealth insurance | Housing allowance | Life insurance | Paid Holidays | Paid sick timeEntry-level InternshipSan Jose, California, United States6h ago
-
Machine Learning Engineer Intern (E-commerce Governance Algorithms) - 2026 Start (PhD) USD 114K-114KC++ | Computer Vision | Go | Java | Language ProcessingHealth insurance | Housing allowance | Life insurance | Paid Holidays | Paid sick timeEntry-level InternshipSeattle, Washington, United States6h ago
-
Machine Learning Engineer - Inference USD 162K-387KDistributed Systems | GPU Acceleration | High Performance | High-Performance Computing | JAX401k savings plan match | Life insurance | Medical, dental, and vision insurance | Paid Holidays | Paid parental leaveMid-level Full TimeSan Jose, California, United States6h ago