Performance Engineer, GPU
San Francisco, CA | New York City, NY | Seattle, WA
USD 280K-850K Senior-level Full Time
Tasks
- Architect GPU performance systems
- Build distributed GPU communication strategies
- Develop custom GPU kernels
- Develop performance modeling frameworks
- Implement GPU utilization optimizations
- Implement kernel fusion strategies
- Improve end to end training and inference efficiency
- Optimize tensor core performance
- Partner with hardware vendors
- Profile production ML performance bottlenecks
Perks/Benefits
- Flexible working hours
- Generous vacation
- Hybrid work 25 percent
- Optional equity donation matching
- Parental leave
- Visa sponsorship support
Skills/Tech-stack
Bandwidth Optimization | CUDA | Cluster Orchestration | Collective communication | Custom Operators | Cutlass | FP8 Quantization | Fault Tolerance | Flash Attention | Int8 Quantization | JAX | Kernel Fusion | Memory bandwidth | Memory bandwidth optimization | Mixed Precision | Model Parallelism | NCCL | NVLink | Nsight | PyTorch | Tensor Core | Tensor core optimization | Torch compile | Triton | XLA
Education
Regions
Countries
States
Related jobs
-
Mid-Level Data Engineer USD 90K-98KAPI Development | Azure Data | Azure Data Factory | Azure Data Lake | Azure Data Lake StorageRemote workMid-level Full TimeWork from home, VA, United States R10h ago
-
Senior Data Engineer USD 165K-180KAPIs | Anomaly Detection | Azure | Azure Data | Azure Data FactorySenior-level Full TimeWork from home, VA, United States R10h ago
-
Evergreen - Mathematics for Machine Learning USD 80K-300KAutodiff | JAX | Linear Algebra | Matrix Operations | NumPyAsynchronous hiring process | Flexible collaboration | Part-time hoursMid-level Full TimeBoston, US13h ago
-
Staff Software Engineer, ML Frameworks USD 207K-300KAPIs | Data Processing | Debugging | Fine Tuning | GPU AccelerationSenior-level Full TimeMountain View, CA, USA15h ago
-
AWS | Artificial Intelligence | Azure AI | Data Analysis | DatabricksBackup childcare | Financial coaching | Health care coverage | Mental health support | On-site health and wellness centersSenior-level Full TimeChicago, IL, United States1d ago
-
Staff Machine Learning Engineer (Pricing) USD 215K-322KA/B Testing | API Design | AWS | Automated retraining | B testing401k matching | Dental insurance | Family planning assistance | Flexible time off | Healthcare benefitsSenior-level Full TimeSan Francisco, CA1d ago
-
AWS | C++ | EC2 | Git | GoSenior-level Full TimeNew York, United States1d ago
-
AI Engineer SME USD 169K-229KAWS GovCloud | Azure Government | Bias Mitigation | Cloud Computing | Computer Vision401k match | Dental insurance | Disability insurance | Flexible work weeks | Health insuranceSenior-level Full TimeUSA HI Honolulu - 155 Kapalulu …1d ago
-
AI Solutions Engineer, East USD 125K-175KAWS | Azure | Cloud platform | Dspy | Generative AI401k plan | Dental insurance | Medical insurance | Mental wellness support | Parental leaveMid-level Full TimeRemote (New York) R1d ago
-
Machine Learning Engineer, Relevance and Personalization USD 157K-185KA/B | A/B Testing | Apache Airflow | Apache Kafka | Apache SparkMid-level Full TimeUnited States1d ago
-
Senior Machine Learning Engineer USD 160K-215KAWS Batch | AWS S3 | Amazon EC2 | Amazon ECR | Camera CalibrationDental insurance | Equity package | Health insurance | Retirement savings planSenior-level Full TimeCosta Mesa, CA1d ago
-
Machine Learning Engineer USD 104K-131KAWS | Active Learning | Airflow | Argo | Azure401k match | Corporate discounts | Education assistance | Flexible work options | Maternity leaveSenior-level Full TimeRemote, REMOTE, United States R1d ago
-
Software Engineer III - AI/ML USD 176K-185KAI Platform | AWS SageMaker | Agile | Airflow | Azure MLBackup childcare | Financial coaching | Health care coverage | Mental health support | On-site health and wellness centersSenior-level Full TimePlano, TX, United States1d ago
-
Machine Learning Systems Engineer USD 144K-192KCUDA | Data loading | Distributed Training | Gradient Computation | Kernel Fusion401k match | Dental insurance | Health Accounts | Health insurance | Health savings accountSenior-level Full TimeBoston, Massachusetts, United States R1d ago
-
Machine Learning Scientist - Natural Language Processing (NLP) - Vice President - Machine Learning Center of Excellence USD 195K-260KA/B | A/B Testing | B testing | Cloud Computing | Code debuggingBackup childcare | Financial coaching | Health care | Mental health support | Onsite health and wellness centersExecutive-level Full TimeSeattle, WA, United States1d ago
-
Software Engineer III- Python, PySpark, ETL, AWS USD 170K-185KAWS | AWS Lambda | Agile | Amazon EMR | Amazon S3Senior-level Full TimeJersey City, NJ, United States1d ago
-
Senior Robotics Software Engineer USD 150K-199KC++ | CUDA | Collision detection | Computer Vision | LinuxDental insurance | Medical insurance | Paid time off | Vision insuranceSenior-level Full TimeOakland, CA1d ago
-
Senior Machine Learning Engineer, Trust USD 200K-235KA/B | A/B Testing | B testing | Batch Processing | Boosted TreesEmployee travel credits | Remote eligibleSenior-level Full TimeSan Francisco, CA1d ago
-
Machine Learning Systems Engineer USD 144K-192KCUDA | Kernel Fusion | NVIDIA Nsight | PyTorch | PyTorch Profiler401k match | Dental insurance | Health insurance | Health savings account | Life insuranceSenior-level Full TimeRemote U.S. R1d ago
-
Machine Learning Systems Engineer USD 144K-192KCUDA | Data loading | Distributed Training | Kernel Fusion | NsightMedical Dental Vision 401k with company match Health Savings Account Life Insurance Pet InsuranceSenior-level Full TimeLas Vegas, Nevada, United States R1d ago
-
Machine Learning Systems Engineer USD 144K-192KCUDA | Kernel Fusion | Nsight | Profiling tools | PyTorch401k match | Dental insurance | Health insurance | Health savings account | Life insuranceSenior-level Full TimePittsburgh, Pennsylvania, United States R1d ago
-
Staff Machine Learning Engineer USD 205K-272KAWS | Active Learning | Azure | CI/CD | Cloud Computing401k match | Dental insurance | Health savings account | Life insurance | Medical insuranceSenior-level Full TimeRemote U.S. R1d ago
-
Staff Machine Learning Engineer USD 205K-272KAWS | Active Learning | Azure | CI/CD | CPU401k match | Dental insurance | Health savings account | Life insurance | Medical insuranceSenior-level Full TimeLas Vegas, Nevada, United States R1d ago
-
Staff Machine Learning Engineer USD 205K-272KAWS | Active Learning | Azure | CI/CD | Cloud Computing401k match | Dental insurance | Health benefits | Health savings account | Life insuranceSenior-level Full TimePittsburgh, Pennsylvania, United States R1d ago
-
Staff Machine Learning Engineer USD 205K-272KAWS | Active Learning | Azure | CI/CD | CPU401k match | Dental insurance | Health savings account | Life insurance | Medical insuranceSenior-level Full TimeBoston, Massachusetts, United States R1d ago