Staff Engineer Tech Lead Manager, ML Acceleration
Tasks
- Apply model optimization techniques
- Build and manage ML and infra team
- Collaborate with ML research training platform and product teams
- Define ML acceleration technical vision
- Define team goals priorities and roadmap
- Design and implement scalable ML acceleration solutions
- Implement hardware acceleration strategies
- Lead distributed training performance improvements
- Measure and report acceleration impact
- Optimize data pipeline and data loader
- Provide technical mentorship and career development
Perks/Benefits
- 401k match
- Dental insurance
- Health savings account
- Life insurance
- Medical insurance
- Pet insurance
- Vision insurance
Skills/Tech-stack
AWS | Azure | C++ | Cloud Computing | Distributed Systems | Google Cloud | JAX | MLOps | Machine Learning | Neural Networks | Performance optimization | PyTorch | Python
Education
N/A
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 R25d ago
-
Strategic Analytics Manager (Commercial Insights & ROI) USD 106K-145KData Quality | Data Visualization | Excel | Forecasting | LookerDental insurance | Health insurance | Other benefits | Remote-first work environment | Vision insuranceMid-level Full TimeRemote (United States) R13h 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 R17h 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 R17h ago
-
Lead AI Engineer USD 198K-261KAgentic Frameworks | CI/CD | Cloud Computing | Containers | Fine TuningSenior-level Full TimeUSA, USA R17h 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 … R19h ago
-
Agent Orchestration | Agentic AI | Artificial Intelligence | Autogen | Autonomous ReasoningSenior-level Full TimeRemote-USA R19h 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 R19h 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 R22h ago
-
AWS | Apache Kafka | Apache Spark | Azure | Cloud platformSenior-level Full TimeFlorida; Remote - Massachusetts; Remote - … R22h ago
-
Amazon Web Services | Apache Hadoop | Apache Kafka | Apache Spark | Big DataRemote work | Travel opportunitiesSenior-level Full TimeFlorida; Remote - Illinois; Remote - … R23h ago
-
Lead AI Development Engineer USD 115K-161KAgentic AI | Artificial Intelligence | AutoCAD | Automation | BOMComprehensive benefits package | Fully remote | Limited travelSenior-level Full TimeUS - Remote R23h 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
-
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
-
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
-
NLP AI Engineer USD 130K-180KAWS | Cloud platform | DeepSpeed ZeRO | Direct Preference Optimization | Distributed Training100 percent remote U.S. | Direct W2Senior-level Full TimeUnited States - Remote R1d ago
-
Senior-level Full TimeUnited States - Remote R1d ago
-
Applied AI Engineer USD 130K-180KAWS | Agentic AI | CI/CD | Cloud platform | Computer VisionCareer growth | Remote workSenior-level Full TimeUnited States - Remote R1d ago
-
Embedded Robotics Engineer USD 125K-145KC++ | Concurrent Systems | Control Systems | Debugging | GazeboCareer growth | Code review | Design review | Mentorship | Remote workSenior-level Full TimeUnited States - Remote R1d ago
-
Embedded Software Engineer USD 130K-195KARM Cortex | ARM Cortex A | ARM Cortex-M | Agile | Azure DevOpsSenior-level Full TimeUnited States - Remote R1d ago
-
Quantitative Engineer USD 150K-185KC++ | Cloud Native | Cloud Native Architecture | Columnar Databases | ConcurrencyCareer growth | Mentorship | Remote workSenior-level Full TimeUnited States - Remote R1d ago
-
Senior-level Full TimeUSA- New York Remote Office, United … R1d ago