Staff ML Engineer, Generative Model Performance & Efficiency
Mountain View, California, United States, New York City, New York, United States
USD 251K-310K Senior-level Full Time
Tasks
- Analyze model architectures for training performance bottlenecks
- Build performance profiling and debugging tools
- Design low latency high throughput serving systems
- Develop quantization and model compression techniques
- Implement model partitioning and sharding strategies
- Optimize model code for TPUs and GPUs
- Optimize training and inference performance
Perks/Benefits
- N/A
Skills/Tech-stack
Data parallelism | Diffusion Models | Efficient Attention | Expert parallelism | Flax | GPU | High Throughput | JAX | Knowledge Distillation | Low Latency | Low-latency serving | Mixture of Experts | Model Compression | Model partitioning | NVIDIA Nsight | Perfetto | Pipeline parallelism | Pruning | PyTorch | Python | Quantization | TPU | Tensor Parallelism | TensorFlow | Transformers | XLA | Xprof
Education
Regions
Countries
States
Related jobs
-
Associate Director, AI Enablement & Machine Learning USD 212K-318KCloud Computing | Data Engineering | Data Governance | Deep learning | Generative AIAnnual incentive program | Healthcare coverage | Retirement benefitsMid-level Full TimeCambridge, MA9h ago
-
Software Engineer, Data (L1) USD 75K-130KAI Platform | APIs | Apache Airflow | Apache Iceberg | Apache Spark401k match | Calm subscription | Career growth and learning opportunities | Commuter benefits options | Dental insuranceMid-level Full TimeAustin, TX11h ago
-
Solutions Engineer – Robotics & Autonomous Driving USD 152K-500K2D Imaging | 3D Imaging | Autonomous Driving | C++ | CameraCommuter stipend | Generous PTO | Global travel | Health, dental, vision coverage | Learning and development stipendSenior-level Full TimeSan Francisco, CA; New York, NY12h ago
-
Senior Software Engineer, Storage USD 217K-303KC++ | Caching | Cassandra | Data Storage | Distributed Systems401k employer match | Dental insurance | Equity compensation | Generous time off | Medical insuranceSenior-level Full TimeRemote - United States R14h ago
-
Machine Learning Research Engineer USD 161K-189KAWS | Azure | Bias Variance | Bias-Variance Tradeoff | C++Flexible hybrid work model | Mental health counseling | Mentorship programs | Paid parental leave | Paid volunteer time offMid-level Full TimeNew York, US, New York15h ago
-
Sr. Application Software Engineer, Data Analytics USD 160K-225KAngular | C# | CI/CD | Computer Vision | Continuous integrationExtended hours | Travel | Weekend workSenior-level Full TimeBastrop, TX16h ago
-
AI Engineer - Automation/Robotics USD 104K-110KAlgorithms | Azure Machine Learning | Computer Vision | Data Structures | DebuggingOnsite workEntry-level Full TimeRedmond, WA16h ago
-
Mid-level Full TimeEl Segundo17h ago
-
Senior-level Full TimeSan Francisco, California17h ago
-
Foundation Model Data, Software Engineer USD 213K-263KApache Beam | Apache Flume | C++ | CI/CD | Distributed Systems401k match | Baby bonding leave | Dental insurance | Disability insurance | Health insuranceSenior-level Full TimeMountain View, CA, USA; San Francisco, …17h ago
-
Senior-level Full TimeOnsite - Austin, TX17h ago
-
Machine Learning Engineer USD 196K-196KApache Airflow | CI/CD | Containerization | Docker | Experiment trackingFully remoteSenior-level Full TimeSan Francisco, CA R17h ago
-
Lead Cyber Risk & Analytics Engineer USD 130K-160KAgile | Data Visualization | Database querying | Econometrics | ExcelDependent care assistance | Flexible hours | Hybrid work | Learning and development | Mentorship programsSenior-level Full TimeNew York Office17h ago
-
Software Engineer - Embedded Firmware USD 125K-210KAvionics | C Programming | C++ | Communications Systems | DebuggingDental insurance | Equity compensation | Medical insurance | Paid time off | Performance bonusesSenior-level Full TimeSouth San Francisco, California, USA17h ago
-
AWS Bedrock | Agent systems | Anthropic API | Autogen | Azure401k matching program | Adoption Assistance | Development and career growth opportunities | Fertility treatments | Flexible work schedulesSenior-level Contract Full TimeRemote, OR, United States R18h ago
-
Senior Data Engineer USD 90K-110KAWS | Agile | Apache NiFi | Data Architecture | Data ModelingAutonomy | Flexible working hours | Global employee assistance programme | Online training videos | Teambuilding eventsSenior-level Full TimeNew York, United States18h ago
-
Data Platform Engineer USD 85K-120KAlembic | App Service | Azure | Azure App | Azure App Service401k match | Annual discretionary bonus program | Bereavement leave | Commuter spending accounts | Dental insuranceMid-level Full TimeChicago, IL (Downtown)18h ago
-
Robotics Test Engineer USD 100K-150KASan | Bitbucket Pipelines | C++ | CI/CD | DebuggingDirect hire | Mentorship | Ownership of test suitesMid-level Full TimeOnsite - Austin, TX19h ago
-
AI Engineer 1 - Compliance, Quality & Testing USD 80K-150KAI-assisted testing | CI/CD | Code review | Cypress | Defect Tracking401k match | Dental insurance | Holidays | Medical insurance | Paid time offEntry-level Full TimeWashington, DC19h ago
-
Machine Learning Engineer, ML Flywheel USD 136K-173KC++ | Data Pipelines | Distributed Systems | ML Ops | Machine LearningAnnual bonus program | Company benefits program | Equity incentive plan | Hybrid workMid-level Full TimeMountain View, CA, USA20h ago
-
Software Engineer, Applied AI USD 190K-280KContext engineering | Data Processing | Data Storage | Debugging | Docker401k match | Dental insurance | Health insurance | Hybrid work model | Professional developmentSenior-level Full TimeNew York City R20h ago
-
Data Engineer USD 74K-133KAgile | Apache Airflow | BigQuery | Cloud Composer | Cloud Data401k retirement plan | Dental insurance | Disability insurance | Flexible time off | Health insuranceMid-level Full TimeLisle, IL, United States R20h ago
-
AI Data Engineer USD 133K-185KAmazon Redshift | BigQuery | CI/CD | CRM Integration | Clay401(k) self contribution | BetterUp coaching | Company holidays observed | Development plan | Flexible paid time offSenior-level Full TimeAustin, TX21h ago
-
Senior Staff Data Engineer - Platform Data and Analytics USD 268K-368KAirflow | Alerting | Apache Spark | Capacity Planning | Cost OptimizationSenior-level Full TimeSan Francisco, CA R21h ago
-
Staff Data Engineer USD 185K-220KAWS | Apache Airflow | Apache Kafka | Benthos | Big DataDental insurance | Disability insurance | Flexible work hours | Health insurance | Health savings accountSenior-level Full TimeRosslyn, VA or Remote R21h ago