Staff Software Engineer, AI/ML Performance
Tasks
- Achieve state of the art performance for customer launches
- Co design TPU friendly models
- Drive performance improvements using benchmarks
- Identify and maintain ML training and serving benchmarks
- Participate in algorithmic innovation for TPU hardware features
Perks/Benefits
- N/A
Skills/Tech-stack
Algorithm Design | Auto sharding | C++ | Code generation | Compiler optimization | Custom Kernels | Data Structures | Data Structures and Algorithms | Debugging | GPU | Language Processing | Machine Learning | Memory hierarchy | Model Quantization | Natural Language | Natural Language Processing | Performance optimization | Python | Runtime Systems | Software Architecture | Sparsity | TPU
Education
Bachelor of Engineering | Bachelor of Science | Master of Science | PhD
Roles
Regions
Countries
States
Cities
Related jobs
-
Data Engineer USD 192K-196KBig Data | Data Architecture | Data Visualization | Data Warehouse | Database systemsEntry-level Full TimeMenlo Park, CA3h ago
-
Data Engineer, Analytics USD 209K-235KBig Data | Data Governance | Data Modeling | Data Quality | Data VisualizationSenior-level Full TimeMenlo Park, CA3h ago
-
Data Engineer, Analytics USD 185K-196KBig Data | Data Governance | Data Quality | Data Security | Data WarehousingMid-level Full TimeMenlo Park, CA3h ago
-
Senior-level Full TimeNew York, NY3h ago
-
Accessibility | Artificial Intelligence | Data Analysis | Data Mining | Data ProcessingSenior-level Full TimeSunnyvale, CA, USA4h ago
-
Software Engineer III, Embedded Systems Firmware, Platforms Infrastructure Engineering USD 147K-211KC# | C++ | Data Structures | Data Structures and Algorithms | Device DriversSenior-level Full TimeSunnyvale, CA, USA4h ago
-
Technical Lead Manager, ML and GenAI, Ads Safety USD 207K-300KAgentic AI | Artificial Intelligence | Context learning | Distributed Computing | EmbeddingsSenior-level Full TimeMountain View, CA, USA; New York, …4h ago
-
API Integration | Cloud Architecture | Data Processing | Deep learning | GPUTravel up to 20 percent timeSenior-level Full TimeReston, VA, USA; Washington D.C., DC, …4h ago
-
AI Middleware Engineer USD 99K-195KAI Agent | AI Agent Frameworks | API Design | API Security | Agent FrameworksClient-facing work | Production impactMid-level Full TimeSan Francisco10h ago
-
Software Engineer, Applied AI USD 130K-500KData Pipelines | Data Quality | Evaluation | Experimental Design | GoEquity 4 year vest | Free gym membership | Health insurance | Housing bonus | Meal stipendMid-level Full TimeSan Francisco12h ago
-
Senior Data Engineer USD 136K-160KApache Spark | Azure Data | Azure Data Factory | Azure Event | Azure Event Hubs401k match | Employee assistance program | Free parking | Healthcare | Hybrid workplace flexibilitySenior-level Full TimeCharlotte, North Carolina, United States; Virtual R13h ago
-
Optimization Software Engineer USD 132K-198KAPIs | AWS | Algorithms | Amazon CodeWhisperer) | Azure401k matching | Commuter benefits | Healthcare benefits | Life and disability insurance | Mental health resourcesMid-level Full TimeWashington, District of Columbia, United States13h ago
-
Quantum Software Engineer USD 132K-198KC# | C++ | CI/CD | CUDACUDAQ | CUDALabCaregiver leave | Commuter benefits | Disability insurance | Family planning support | Life insuranceMid-level Full TimeWashington, District of Columbia, United States13h ago
-
Analog Sensors | Bare Metal | Best practices | Board Bring-up | Bring-up401k employer match | Caregiver & wellness leave | Commuter benefits | Comprehensive medical, dental, vision | Generous time offSenior-level Full TimeCosta Mesa, California, United States14h ago
-
Mid-level Full TimeDes Moines, Iowa15h ago
-
Senior-level Full TimeNew York, NY, United States15h ago
-
Robotic Software Engineer - Air Autonomy USD 135K-175KAgile | ArduPilot | Bash | C++ | CI/CDIn-office work flexibility | Occasional remote work | Paid time off | Travel opportunitiesMid-level Full TimeWashington, District of Columbia, United States16h ago
-
Robotic Software Engineer - Aerial Autonomy USD 125K-165KAgile | ArduPilot | Bash | C++ | CI/CDTravel for site integration and demonstrationsMid-level Full TimeFort Walton Beach, Florida, United States16h ago
-
Robot Learning Engineering Intern USD 80K-116KBehavior Cloning | Computer Vision | Force control | Imitation Learning | Learning from DemonstrationCatered lunches | Employee Assistance Program (EAP) | Flexible PTO | Healthy snacks | Paid sick leaveEntry-level InternshipOnsite- Pittsburgh, PA16h ago
-
Product Enablement Data Engineer USD 112K-182KAWS | Authentication | Azure | Curl | DNS401k | Dental insurance | Disability benefits | Flexible time off | Health insuranceSenior-level Full TimeUSA17h ago
-
AI Data Engineer USD 133K-256KAWS | Agile | Apache Kafka | Apache Spark | AzureHybrid work | Work from home up to three days per weekEntry-level Full TimeMaryland, United States17h ago
-
Sr Machine Learning Engineer I USD 150K-241KAutomated retraining | Bayesian Filter | CI/CD | Cloud Computing | Data AnalysisDiscretionary paid time off | Emotional and mental wellness support | Employee resource groups | Fitness programs | Learning and development programsSenior-level Full TimeSeattle, Washington, United States17h ago
-
AWS Lambda | Amazon Bedrock | Amazon ECR | Amazon S3 | Amazon SageMaker401k match | Dental insurance | Health insurance | Long-term engagement | Remote-firstSenior-level Full TimeUSA17h ago
-
BACnet | BMS | Control Systems | Control loops | Cooling systemsFlexible working hours | Generous vacation | Optional equity donation matching | Parental leave | Visa sponsorship assistanceMid-level Full TimeRemote-Friendly, United States R18h ago
-
AI/ML Research Engineer USD 120K-220KAWS | Cloud Computing | Computational Biology | Data Processing | Data pipelineMid-level Full TimeBoston18h ago