Engineering Manager, ML Performance
USD 207K-300K Senior-level Full Time
Tasks
- Achieve performance for customer launches
- Collaborate on custom kernel development
- Drive compiler and runtime optimizations
- Identify performance opportunities using benchmarks
- Lead software engineers on ML training and serving benchmarks
- Participate in TPU hardware co design
- Support model preserving optimizations
Perks/Benefits
- N/A
Skills/Tech-stack
Auto sharding | Auto-tuning | CUDA | CUDA programming | CUDA runtime | Compiler optimization | Computer Architecture | Data Processing | Debugging | Distributed Systems | Fine Tuning | GPU | JAX | Kernel optimization | LoRA | ML benchmarking | MLIR | Machine Learning | Model Deployment | Model Evaluation | Model Fine-tuning | OpenXLA | Performance Analysis | PyTorch | Quantization | Sparsity | Speculative decoding | TPU | Triton | VLLM
Education
Bachelor of Engineering | Bachelor of Science | Master of Science | PhD
Regions
Countries
States
Related jobs
-
Associate Director, Applied AI Engineering USD 130K-268K.NET | A/B | A/B Testing | AI Agent | AI agent orchestrationMid-level Full TimeDallas, Texas, United States; Hermitage, Tennessee, …4h ago
-
Associate Director, Applied AI Engineering - PxE ERM USD 130K-268K.NET | A/B | A/B Testing | AI Agent | AI agent orchestrationMid-level Full TimeDallas, Texas, United States; Hermitage, Tennessee, …4h ago
-
Associate Director, Applied AI Engineering-PxE Platforms USD 130K-268K.NET | AI Agent | AI agent orchestration | API Gateway | AWS BedrockMid-level Full TimeDallas, Texas, United States; Hermitage, Tennessee, …4h ago
-
.NET | AI Agent | AI agent orchestration | API Gateway | AWSMid-level Full TimeDallas, Texas, United States; Hermitage, Tennessee, …4h ago
-
Agent systems | Artificial Intelligence | Backend Development | Cloud Computing | ComplianceSenior-level Full TimeNew York, NY, USA6h ago
-
Software Engineer Manager II, Google Cloud Storage USD 207K-300KCode review | Compute Technologies | Distributed Systems | Infrastructure Development | Networking401k match | Baby bonding leave | Dental insurance | Disability insurance | Health insuranceSenior-level Full TimeSeattle, WA, USA6h ago
-
Manager, Software Engineering - Data Platform USD 258K-376KBatch Processing | Cross-Functional Collaboration | Cross-functional | Data Infrastructure | Data platformMid-level Full TimeSan Francisco, CA • New York, … R10h ago
-
Director, Field Engineering (CPG & Retail) USD 262K-360KAnalytics | Artificial Intelligence | Cloud | Customer Relationship Management | Customer relationshipExecutive-level Full TimeUnited States15h ago
-
Tech Lead Manager, Cortex AI - Snowflake Intelligence USD 200K-287KData Modeling | Experimentation | Function Calling | Go | Information RetrievalSenior-level Full TimeUS-CA-Menlo Park17h ago
-
AI Agents | Agent Orchestration | Agent systems | Backend Engineering | Distributed SystemsDental insurance | Health insurance | In-office lunch | Unlimited PTO | Visa sponsorship supportExecutive-level Full TimeSan Francisco, CA; Onsite17h ago
-
Business Analysis | Cloud Cost Control | Cost Control | Customer Success | Data and AIHybrid work | Travel up to 30%Senior-level Full TimeUnited States R17h ago
-
AI | Apache Cassandra | Caching | Capacity Planning | Data ModelingField testing opportunities | Hybrid work model | Offsite collaborationMid-level Full TimeSan Francisco, CA; Los Angeles, CA; …17h ago
-
API configuration | Accounting workflows | Change Management | Data Governance | Data Mining401k savings plan | Dental insurance | Dependent Ski Pass Programs | Employee Ski Pass | Flexible spending accountSenior-level Full TimeAspen, Basalt or Denver, Colorado, United …19h ago
-
Agile | Big Data | Cloud Computing | Deployment | DesignMid-level Full TimeAustin; Sunnyvale21h ago
-
Agile | Big Data | Cloud Computing | Data platforms | DeploymentMid-level Full TimeAustin; Sunnyvale21h ago
-
Staff Product Manager - AI (Data & Analytics) USD 180K-210KA/B | A/B Testing | APIs | Agentic AI | AnalyticsSenior-level Full TimeAtlanta, GA R1d ago
-
Alerting | Bioinformatics | Cause analysis | Change Control | Data Analysis401k | Baby bonding leave | Commuter benefits | Dental insurance | Disability insuranceMid-level Full TimeUS Remote R1d ago
-
AWS | Amplicon sequencing | Cell Free DNA | Cloud Computing | Containerization401k | Commuter benefits | Employee referral program | Fertility care benefits | Free testingMid-level Full TimeUS Remote R1d ago
-
Director of Software engineer USD 150K-195KAPI Design | Agile | Amazon Web Services | Automation | AzureExecutive-level Full TimeNew York, United States1d ago
-
Product Owner, AI Internal Systems USD 110K-160KAPIs | Artificial Intelligence | Automation | Cloud infrastructure | Database systemsEntry-level Full TimeUnited States R1d ago
-
Director, AI Engineering USD 176K-285KAWS Lambda | AWS SageMaker | Agentic AI | Amazon Bedrock | DockerDental insurance | Medical insurance | Office spaceExecutive-level Full TimeAtlanta, GA1d ago
-
Senior Product Manager, AI Laboratory Analytics USD 160K-200KAnalytics | Artificial Intelligence | Customer interviews | Data Analysis | Decision Support Systems401k savings plan | Paid time off | Parental leave | Remote work | Stock incentive programsSenior-level Full TimeTucson, United States1d ago
-
Launch Manager, Google Play Trust and Safety USD 129K-191KAI Classifiers | Anti-abuse | Artificial Intelligence | Content Moderation | Content policyMid-level Full TimeKirkland, WA, USA; Austin, TX, USA1d ago
-
ML Infrastructure Engineer USD 200K-350KActivation checkpointing | CUDA | Data Pipelines | Data Validation | Dataloader optimizationEntry-level Full TimeRedwood City, CA1d ago
-
Senior Manager, Perception USD 196K-280KAI software | AI software architecture | CUDA | Computer Vision | E2E LearningSenior-level Full TimeFoster City, CA1d ago