Senior / Staff ML Training Optimization Engineer
Tasks
- Build distributed training frameworks
- Create tooling and dashboards
- Design CUDA kernels
- Evaluate emerging training and inference technologies
- Identify performance bottlenecks
- Implement quantization aware training
- Profile model runtime and memory
Perks/Benefits
- Catered meals
- Dental insurance
- Flexible hours
- Health insurance
- Snacks
- Social events
- Team-building activities
- Unlimited vacation
- Vision insurance
- Work from home
Skills/Tech-stack
Bazel | C++ | CPU Profiling | CUDA | CUDA kernels | GPU Profiling | JAX | Kubernetes | NVIDIA Nsight | PyTorch | PyTorch Profiler | Python | Quantization | Rust
Education
Bachelor of Engineering | Bachelor of Science | Master of Science | PhD
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 R5h ago
-
Senior Data Engineer USD 165K-180KAPIs | Anomaly Detection | Azure | Azure Data | Azure Data FactorySenior-level Full TimeWork from home, VA, United States R5h ago
-
Anomaly Detection | Data Modeling | Data Pipelines | Data schemas | Docker100 percent remote work | Autonomous work environment | Career growth opportunities | Flexible work environment | International team collaborationMid-level Full TimeCanada R16h ago
-
Senior Data Engineer USD 129K-165KAWS | Airflow | CI/CD | Data Modeling | Django401k | Half-day Fridays | Medical/Dental/Vision insurance | Paid Holidays | Remote workSenior-level Full TimeChicago, IL, US R19h ago
-
Solutions Engineer USD 150K-180KAI | Apache Flink | Apache NiFi | Apache Spark | Applied ScienceContinued Career Development | Employee resource groups | Flexible work from home | Generous paid time off | Paid volunteer timeMid-level Full TimeUS-California-Remote, United States R21h ago
-
Senior Embedded Software Engineer - Audio USD 115K-150KAIDL | ALSA | AOSP | ASoC | AndroidAdoption support | Adoption/support reimbursement | Child care assistance | Dental insurance | Employee resource groupsSenior-level Full TimeDearborn, MI, United States R23h ago
-
Senior Software Engineer, Storage USD 166K-210KAmazon CloudWatch | Amazon EC2 | Datadog | Go | MemcachedAnnual refresh grants | Equity grant | Remote work flexibilitySenior-level Full TimeUnited States - Remote R23h 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
-
Senior Data Engineer, Sentinel (Pacific Time Zone) USD 153K-210KAWS | Airflow | Alerting | CI/CD | DatabricksSenior-level Full TimeUnited States R1d ago
-
Mid-level Full TimeAnywhere USA, United States R1d 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
-
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
-
AWS S3 | Access Control | Apache Iceberg | Apache Spark | AtlanDental coverage | ESPP | Flexible spending wallets | Health coverage | Remote workSenior-level Full TimeRemote Canada R1d ago
-
AWS EKS | Access Control | Apache Iceberg | Apache S3 | CI/CDDental and vision coverage | ESPP | Flexible spending wallets | Remote-first work | Subsidized medical coverageSenior-level Full TimeRemote US R1d ago
-
Senior Data Engineer - Agentic AI Engineering USD 138K-173KAWS | Access Control | Airflow | Azure | DBTSenior-level Full TimeUnited States of America R1d ago
-
Principal Data Engineer USD 152K-190KApache Spark | Artificial Intelligence | CI/CD | Cloud Platforms | Code Coverage401k company match | Dental insurance | Flexible paid time off | Life insurance | Long-term disabilitySenior-level Full TimeDallas, TX - Hybrid (3x in … R1d ago
-
Software / Computer Science Intern USD 42K-52KData Parsing | Data Pipelines | Debugging | Networking | PythonHybrid work flexibility | Professional developmentEntry-level InternshipMonroeville, PA R1d 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
-
Azure Data | Azure Data Factory | Azure Synapse | Azure Synapse Analytics | CI/CDCollaborative work environment | Dental insurance | Flexible remote working options | Health insurance | Professional development and training opportunitiesSenior-level Full TimeWashington R1d ago
-
Azure Data | Azure Data Factory | Azure Synapse | Azure Synapse Analytics | CI/CDCollaborative work environment | Dental benefits | Health benefits | Impactful national security projects | Professional development and trainingSenior-level Full TimeVirginia R1d ago
-
Azure Data | Azure Data Factory | Azure Synapse | Azure Synapse Analytics | CI/CDCollaborative work environment | Dental benefits | Health benefits | Impactful projects | Professional development and trainingMid-level Full TimePennsylvania R1d ago
-
Azure Data | Azure Data Factory | Azure Synapse | Azure Synapse Analytics | CI/CDDental insurance | Health insurance | Professional development and training | Remote work flexibility | Retirement benefitsSenior-level Full TimeTexas R1d ago
-
Azure Data | Azure Data Factory | Azure Synapse | Azure Synapse Analytics | CI/CDDental benefits | Health benefits | Professional development and training | Remote work flexibility | Retirement benefitsSenior-level Full TimeNorth Carolina R1d ago