Staff Software Engineer, ML Systems Co-Design
Sunnyvale, CA, USA; Kirkland, WA, USA
USD 207K-300K Senior-level Full Time
Tasks
- Adapt open source models to TPU native variants
- Author reference implementations for pre silicon goal reachability
- Deliver performance ladders pre silicon for planning and optimization
- Optimize agentic workflows and reasoning loops for TPU
- Partner with cross functional leaders to steer multi year roadmaps
Perks/Benefits
- N/A
Skills/Tech-stack
Accelerator Kernel Programming | C++ | CUDA | CUDA kernels | Compiler Intermediate Representation | Cycle-accurate simulation | Data Visualization | Debugging | Emulation | HLO | High Performance | High-Performance Computing | Intermediate representation | JAX | Kernel programming | Large-scale | Large-scale Systems | Large-scale systems analysis | MLIR | Machine Learning | Performance Computing | Performance Modeling | PyTorch | PyTorch/XLA | Python | Roofline Models | Simulation | Software Architecture | Software Deployment | Software Design | Systems Analysis | Testing | Triton | XLA
Education
Roles
Regions
Countries
States
Related jobs
-
Featured Feat. Principal Knowledge & Data Architect USD 174K-284KAWS Neptune | Canonicalization | Chunking | Cypher | DBTBenefits including health and wellness programs | Health, wellness, and retirement plansSenior-level Full TimeHeadquarters - Chevy Chase, MD R26d ago
-
Senior-level Full TimeMcLean, VA; McLean, VIRGINIA; McLean, VIRGINIA; …5h ago
-
Senior-level Full TimeMcLean, VIRGINIA5h ago
-
Data Engineer (Expert) WA07 USD 133K-238KData Mapping | Data Modeling | Data Quality | Data Testing | ETLSenior-level Full TimeChantilly, Virginia5h ago
-
Data Engineer (Subject Matter Expert) WA28 USD 133K-238KApache NiFi | CSS | Data Modeling | Data Quality | ETLSenior-level Full TimeChantilly, VA5h ago
-
Data Engineer (Expert) WA12 USD 133K-238KApache NiFi | Data Mapping | Data Modeling | Data Quality | Data TestingSenior-level Full TimeMcLean, VIRGINIA5h ago
-
Data Engineer (Subject Matter Expert) CK16 USD 133K-238KApache NiFi | C# | C++ | Cloud Computing | Data MappingCompetitive benefits packageSenior-level Full TimeMcLean, VIRGINIA5h ago
-
System Engineer- Storage (Expert) TI28 USD 100K-150KActive Directory | Auditing | Backup and Recovery | Bash | Capacity PlanningSenior-level Full TimeMcLean, VIRGINIA5h ago
-
Mid-level Full TimeMcLean, VIRGINIA5h ago
-
API Integration | Cloud Computing | Dashboards | Data Architecture | Data GovernanceMid-level Full TimeFalls Church, VA; Falls Church, VA5h ago
-
Data Engineering | ETL | MongoDB | Object-Oriented | Object-oriented programmingSenior-level Full TimeChantilly, VA; Reston, VA; Chantilly, VA5h ago
-
AI Engineer II - Active TS/SCI w/poly clearance USD 113K-190KC# | Data Engineering | Data Governance | Data Modeling | Data StructuresDental insurance | Disability insurance | Educational assistance | Employee assistance program | Health insuranceMid-level Full TimeAnnapolis Junction, Maryland5h ago
-
Software Engineer 3 – AI/ML USD 113K-279KAWS | Artificial Intelligence | Azure | Docker | ElasticsearchAgile environment collaboration | Partial teleworkSenior-level Full TimeAnnapolis Junction, MD5h ago
-
Senior-level Full TimeLinthicum, MD; Annapolis Junction, MD; Annapolis …5h ago
-
Software Engineer USD 113K-279KApache Pig | Apache Spark | Continuous Monitoring | Data Ingestion | Data QualitySenior-level Full TimeLinthicum, MD; Annapolis Junction, MD; Annapolis …5h ago
-
Database Engineer II USD 124K-208KAWS | Agile | Confluence | Django | Docker401k match | Dental insurance | Health insurance | Immediate 401k vesting | Paid time offSenior-level Full TimeLinthicum, MD5h ago
-
Mid-level Full TimeSuitland, MD; Annapolis Junction, Maryland; Crystal …5h ago
-
Systems Engineer - Modeling and Algorithms USD 112K-160KC# | C++ | Data Analysis | Digital System | Digital system modelingSenior-level Full TimeHuntsville, Alabama; Annapolis Junction, Maryland5h ago
-
Apache Tika | Data Visualization | Language Processing | Machine Learning | MbseMid-level Full TimeAnnapolis Junction, Maryland5h ago
-
Anomaly Detection | Data Engineering | Data Lakes | Data Pipelines | Data ProcessingRegular team sessions | Remote friendly schedule | Team collaborationSenior-level Full TimeRemote, Canada (EST); San Francisco, California R5h ago
-
Data Analysis | Data pipeline | Machine Learning | Multimodal Learning | PythonMid-level Full TimeSan Francisco, California; Toronto, Canada; Redwood … R5h ago
-
Active Learning | Adaptive Sampling | Deep learning | Direct Preference Optimization | Human FeedbackMid-level Full TimeSan Francisco, California; Remote, US or … R5h ago
-
Agile | Cloud Storage | Google Cloud | Google Cloud Storage | GraphQLMid-level Full TimeSan Francisco, California; Remote, Canada, US, … R5h ago
-
Artificial Intelligence | Cloud Computing | Data integration | Generative AI | High AvailabilityMid-level Full TimeNew York City, NY; Toronto, Ontario; … R5h ago
-
AWS | AWS Bedrock | AWS CDK | Anthropic | CelerySenior-level Full TimeKitchener, Ontario; Remote, Canada; Remote, Canada; … R5h ago