Staff Software Engineer, Machine Learning Compilers, Edge TPU
Mountain View, CA, USA; Kirkland, WA, USA
USD 207K-300K Senior-level Full Time
Tasks
- Analyze and improve compiler quality
- Build ML compilers for EdgeTPU
- Co optimize execution across CPU GPU TPU
- Compile ML models for EdgeTPU
- Deploy optimized ML models on EdgeTPU
- Design hardware software interface for EdgeTPU
- Extend ML authoring frameworks for EdgeTPU
- Optimize compilation performance
Perks/Benefits
- N/A
Skills/Tech-stack
Accelerator development | Co-design | Compilation Optimization | Compiler development | Compute architecture | Hardware-Software Co-design | Hardware/software | Inference Optimization | JAX | LLVM | MLIR | Machine Learning | Machine Learning Compilers | Machine Learning Infrastructure | Model Deployment | Model Evaluation | Model Inference | Model Inference Optimization | Model Optimization | Parallelization | PyTorch | Software Co-design
Education
Roles
Regions
Countries
States
Related jobs
-
Senior Software Engineer, AI/ML GenAI, Google Cloud USD 174K-252KC++ | Code Reviews | Computer Vision | Data Processing | Data StorageSenior-level Full TimeSunnyvale, CA, USA1h ago
-
Computer Vision | Data Processing | Debugging | Distributed Computing | Fine TuningSenior-level Full TimeSunnyvale, CA, USA1h ago
-
Staff Software Engineer, AI/ML GenAI, Google Cloud AI USD 207K-300KAlgorithms | Computer Vision | Data Preparation | Data Processing | Data StructuresSenior-level Full TimeSunnyvale, CA, USA1h ago
-
Software Engineer III, AI/ML Computer Vision, AR USD 147K-211KC++ | Computer Vision | Data Processing | Debugging | Image classificationSenior-level Full TimeSan Jose, CA, USA1h ago
-
Staff AI Research Engineer, Large User Models USD 207K-300KAlgorithms | Artificial Intelligence | Attention Mechanisms | BERT | Data StructuresSenior-level Full TimeMountain View, CA, USA1h ago
-
Software Engineer, AI/ML, Google Cloud USD 147K-211KAccelerated Linear Algebra | C++ | Data Processing | Debugging | FP8Mid-level Full TimeMountain View, CA, USA1h ago
-
Sr. Data Engineer USD 127K-164KAgentic AI | DBT | Data Governance | Data Quality | Data SecurityComprehensive benefits | Extensive training initiatives | Group insurance | Growth opportunities | Hospitalization insuranceSenior-level Full TimeBangalore, FL, US, 5600766h ago
-
Research Scientist - LLM USD 225K-400KAudio Models | Evaluation Frameworks | Language Models | Language Processing | Large Language ModelsCommuter reimbursement | Internet reimbursement | Medical/Dental/Vision insurance | Phone bill reimbursement | Relocation providedSenior-level Full TimeSan Francisco Bay Area6h ago
-
Research, Mid-Training USD 225K-380KContext Length Extension | Data Filtering | Data weighting | Deep learning | Distributed TrainingMid-level Full TimeSan Francisco Bay Area7h ago
-
Mid-level Full TimeSeattle9h ago
-
Senior Staff Machine Learning Engineer USD 201K-352KAlgorithms | Code Tuning | Code profiling | Design Patterns | Java401(k) plan match | ESPP | Family leave programs | Flexible spending accounts | Flexible time awaySenior-level Full TimeSanta Clara, CALIFORNIA, United States12h ago
-
Staff Computer Vision and Robotics Engineer USD 159K-309K3D Object Recognition | 3D data | 3D data processing | AWS | AgileSenior-level Full TimeMountain View, USA13h ago
-
Senior Staff AI Research Engineer, AI Platform USD 236K-295KAWS | Azure | Benchmarking | Fine Tuning | GCPEducational reimbursement | Employee insurance plans | Unlimited vacation | Wellness reimbursementSenior-level Full TimeNew York, NY13h ago
-
Agent Orchestration | CI/CD | Chunking Strategies | Context engineering | Databricks401k match | Counseling membership | Dental insurance | Flexible time off | Health insuranceMid-level Full Time-REMOTE, USA- R14h ago
-
2 Tower Architecture | Bias Mitigation | Cross-team | Cross-team Collaboration | EthicsSenior-level Full TimeSan Francisco, CA, US; Remote, CA, … R15h ago
-
AI Automation Infrastructure Engineer USD 100K-176KAPI Development | APIs | Agent Frameworks | Anomaly Detection | AnsibleMid-level Full TimeUSA - Sandy, UT15h ago
-
Software Engineer, Data Assembly USD 135K-180KAWS | Bash | C++ | Computer Vision | Data Integrity401k matching | Annual Getaway | Disability insurance | Equity | Game nightsMid-level Full TimeDenver, Colorado15h ago
-
Data Engineer III USD 122K-175KApache Spark | Automated testing | Batch Processing | Data Catalog | Data DriftSenior-level Full TimeDoral, FL16h ago
-
Staff Software Engineer, Model Data Platform USD 250K-334KC++ | Data Compliance | Data Governance | Data Mining | Distributed SystemsBonus program | Company benefits program | Equity incentive planSenior-level Full TimeMountain View, CA USA; San Francisco, …16h ago
-
Machine Learning Engineer USD 120K-160KAdaptive modulation | Anomaly Detection | Beamforming | C++ | Cognitive radioHybrid work scheduleMid-level Full TimeLos Angeles CA R16h ago
-
AI/ML Research Engineer USD 120K-205KAWS | Data Engineering | Deep learning | Distributed Computing | DockerOnsite role | Relocation assistance (if needed)Mid-level Full TimeBoston, MA or San Francisco, CA17h ago
-
Lead Machine Learning Engineer USD 172K-276KAWS | Azure | CI/CD | Databricks | Distributed SystemsRemote workSenior-level Full TimeChicago, Illinois, USA R17h ago
-
Principal AI Engineer USD 115K-160KAPIs | Artificial Intelligence | Backend Engineering | Data Pipelines | Distributed SystemsFlexible work options | No sponsorship for work authorization | Paid time off | Relocation assistance not availableSenior-level Full TimeDallas, TX, United States17h ago
-
Staff AI Research Engineer, Perception USD 170K-291KC++ | Computer Vision | Depth Estimation | Distributed Training | Embedded Systems401k plan | Dental insurance | Equity programs | Fully stocked kitchen | Green card supportSenior-level Full TimeMilpitas, CA18h ago
-
Customer Analytics Intern USD 48K-64KAPIs | Application development | Artificial Intelligence | Dashboard Development | Data PipelinesEmployee resource groups | Flexible scheduling | Inclusive work environmentEntry-level InternshipCambridge, MA18h ago