ML Performance Engineer
Tasks
- Build and maintain benchmark suites and regression frameworks
- Collaborate with ML and platform teams to embed performance best practices
- Document performance tuning playbooks and share findings
- Drive compiler level optimizations with Triton XLA TorchInductor and TVM
- Evaluate new hardware and software and advise adoption
- Identify and eliminate bottlenecks in data loading model compute communication and memory
- Implement and tune quantization sparsity and pruning strategies
- Optimize KV cache continuous batching and speculative decoding
- Optimize data pipelines sharding and storage access patterns
- Optimize distributed training using tensor parallelism pipeline parallelism FSDP and ZeRO sharding
- Optimize training and inference pipelines for throughput latency and cost
- Tune attention implementations using FlashAttention and paged attention
Perks/Benefits
- N/A
Skills/Tech-stack
Attention Mechanisms | Benchmarking | C++ | Compiler optimization | Continuous batching | Deep learning | Distributed Training | FSDP | FlashAttention | GPU Architecture | Inference Optimization | KV cache | Memory Management | Model Parallelism | Paged Attention | Pipeline parallelism | Profiling | Pruning | Python | Quantization | Sparsity | Speculative decoding | TVM | Tensor Parallelism | TorchInductor | Triton | XLA | Zero
Education
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
-
Edge Computing AI Engineer USD 100K-105KC++ | DSP | Edge inference | Embedded Systems | Federated LearningCareer growth | Remote workMid-level Full TimeUnited States - Remote R18h ago
-
Hadoop Big Data Developer USD 100K-150KAWS EMR | Airflow | Apache Atlas | Apache Flink | Apache HudiCareer growth | Direct W2 employment | Remote workSenior-level Full TimeUnited States - Remote R18h ago
-
Machine Learning Data Engineer USD 80K-100KApache Beam | CI/CD | Data Lineage | Data Modeling | Data QualityDirect W2 employment | Remote workSenior-level Full TimeUnited States - Remote R18h ago
-
Data Platform Engineer USD 100K-150KAutoscaling | C++ | Capacity Planning | Distributed Systems | FinOpsRemote workSenior-level Full TimeUnited States - Remote R18h ago
-
Robot Application Developer USD 100K-150KC++ | Computer Vision | Control Systems | Gazebo | IMUCareer growth | Remote workSenior-level Full TimeUnited States - Remote R18h ago
-
Kafka Platform Engineer USD 130K-180KAWS MSK | Access Control | Access Control Lists | Access Control Lists ACLs | AnsibleSenior-level Full TimeUnited States - Remote R18h ago
-
AWS | Azure | Cloud platform | Direct Preference Optimization | Fine TuningComprehensive benefits package | Remote work | Travel to enterprise customers | Visa sponsorship possibleMid-level Full TimeSan Mateo R20h ago
-
Lead AI Engineer- GTM Systems USD 131K-176KAWS Bedrock | AWS Lambda | Agent coordination | Async event-driven architecture | Async/event-drivenNo relocation assistance not provided | Remote workSenior-level Full TimeRemote - US R20h ago
-
Lead AI Engineer USD 198K-261KAgentic Frameworks | CI/CD | Cloud Computing | Containers | Fine TuningSenior-level Full TimeUSA, USA R20h ago
-
A/B | A/B Testing | AWS | AWS SageMaker | Apache Spark401k employer match | Commuter benefit | Dental insurance | Dependent Care Flexible Spending Account FSA | Flexible time offSenior-level Full TimeLos Angeles, California, United States; San … R22h ago
-
Senior Data Scientist, Applied ML USD 154K-200KAWS | Airflow | Apache Spark | DVC | Data Transformation401k employer contribution | Employer-paid life insurance | Flexible Remote Friendly Work Options | Generous paid holidays | Health Savings Account employer contributionSenior-level Full TimeAustin, Texas | Remote R22h ago
-
Staff Engineer Tech Lead Manager, ML Acceleration USD 172K-229KAWS | Azure | C++ | Cloud Computing | Distributed Systems401k match | Dental insurance | Health savings account | Life insurance | Medical insuranceSenior-level Full TimeRemote U.S. R23h ago
-
Sr Machine Learning Engineer I USD 150K-241KAutomated retraining | Bayesian Filter | CI/CD | Data Association | Extended KalmanDiscretionary paid time off | Emotional and mental wellness support | Employee resource groups | Fitness programs | Learning and development programsSenior-level Full TimeSeattle, Washington, United States R1d ago
-
AWS | Apache Kafka | Apache Spark | Azure | Cloud platformSenior-level Full TimeFlorida; Remote - Massachusetts; Remote - … R1d ago
-
Amazon Web Services | Apache Hadoop | Apache Kafka | Apache Spark | Big DataRemote work | Travel opportunitiesSenior-level Full TimeFlorida; Remote - Illinois; Remote - … R1d ago
-
AI Engineer II ( AI Platform) USD 104K-177K.NET | AWS Bedrock | Asynchronous processing | Audit trails | Azure OpenAIMentorship | Remote work | Soft skill sessions | TrainingMid-level Full TimeUS Remote R1d ago
-
Analytic Server Modeler USD 150K-170KAzure | Big Data | C# | C++ | DDL401k | Medical/Dental/Vision insurance | Paid time offEntry-level Full TimeRemote, United States R1d ago
-
Principal AI Engineer USD 130K-197KAI monitoring | Cloud Computing | Deep learning | Fine Tuning | Language Processing401k match | Employee awards | Hybrid work | Industry certifications | MentorshipSenior-level Full TimeUnited States R1d ago
-
Senior Software Engineer (Storage) - remote in the US USD 100K-155KArgoCD | CSI | Cluster API | Flux | GPU DirectConference attendance | Professional development and trainingSenior-level Full TimeRemote, USA, United States R1d ago
-
Data Engineer - AWS/Databricks USD 122K-210KACID | AWS Glue | AWS S3 | Adaptive Query | Adaptive query executionComprehensive benefits | Mentorship | Training budget | Work-life balanceEntry-level Full TimeReston, VA, United States R1d ago
-
SkillBridge – US 2026 Active Duty US Military ONLY USD 110K-176KApplication development | Data Visualization | Databases | Databricks | Machine LearningRemote workEntry-level Full TimeAtlanta, GA - Remote R1d ago
-
Agentic AI Architect - – Google Gemini Enterprise CX USD 150K-210KAPI Gateway | Apigee | Cloud Functions | Cloud Run | DataflowFinancial and income protection benefits | Medical/Dental/Vision insurance | Paid time off | Tax-advantaged health care accounts | Wellness time offSenior-level Full TimeAustin, TX R1d ago
-
Senior-level Full TimeRemote, United States R1d ago
-
AI-Driven DevOps Engineer USD 132K-198KAI Agent | Alerting | Ansible | Artificial Intelligence | BashAdditional paid day off per month | Comprehensive onboarding and training | Conference participation fee coverage | English language classes discount | Fully remote workSenior-level Full TimeCyprus - Remote R1d ago