Senior Research Engineer, LLM Training & Post-Training
USD 165K-310K Senior-level Full Time
Tasks
- Build and improve PyTorch based training infrastructure and tooling
- Collaborate with customers to translate real world workloads into model improvements
- Contribute to open source projects with features tooling improvements and documentation
- Design build and optimize training pipelines for large language models
- Design evaluation methodologies and benchmark models and analyze failure modes
- Improve model quality using supervised fine tuning continued pretraining preference optimization reinforcement learning and evaluation
- Investigate model training issues convergence instability communication overhead and performance bottlenecks
- Optimize distributed training across multi GPU environments
- Partner with research and engineering teams to build production ready AI systems
Perks/Benefits
- 401k matching
- Company holidays
- Comprehensive health coverage
- Dental coverage
- Flexible time off
- Floating holidays
- Hybrid work model
- In-office meals
- Paid parental and family leave
- Pension contributions
- Professional development allowance
- Sabbatical program
- Two weeks winter break
- Vision coverage
- Wellness benefits
- Work from home stipend
Skills/Tech-stack
CUDA | DPO | DeepSpeed | Distributed Training | Experiment design | FSDP | Fine Tuning | GPU Performance | GPU Performance Optimization | Hugging Face | Hugging Face Transformers | Human Feedback | Learning from Human Feedback | Lightning Fabric | Megatron-LM | Memory Optimization | Mixed Precision | Model Evaluation | Multi-GPU | PEFT | PPO | Performance optimization | Preference optimization | PyTorch | Python | RLHF | Reinforcement Learning | Reinforcement Learning from Human Feedback | Reward Modeling | Supervised Fine Tuning | TRL | TensorRT | Transformers | Triton | VLLM
Education
Regions
Countries
States
Related jobs
-
(US) Principal ML System Engineer USD 139K-304KAWS | Access Control | Alerting | Audit Logging | AuthenticationSenior-level Full TimeRemote, USA R3h ago
-
Senior Software Engineer (Storage) - remote in the US USD 100K-170KArgoCD | CSI | Cluster API | Flux | GitOpsCompany outings and happy hours | Conferences and working groups | Hackathons and tech talks | Professional development and trainingSenior-level Full TimeRemote, USA, United States R10h ago
-
Autonomy Algorithms Software Engineer USD 115K-145KAgent systems | Agile | Algorithms | C++ | CI/CDHybrid work environment | Remote work optionMid-level Full TimeArlington, VA R10h ago
-
Autonomy Algorithms Software Engineer USD 115K-145KAgent systems | Agile | Algorithms | C++ | CI/CDFlexible work schedule | Hybrid work environment | Remote work availableMid-level Full TimeWoburn, MA R10h ago
-
Software Engineer, Embedded Agentic AI USD 170K-210KC# | C++ | Cloud Platforms | Containerization | DockerDisability benefits | Equity awards | Health insurance | Life insurance | Paid time offEntry-level Full TimeSan Jose, California R11h ago
-
Specialist Solutions Architect - Data Engineering & Warehousing (Digital Native Business) USD 180K-247KAWS | Anomaly Detection | Apache Spark | Cloud platform | Data ModelingContinuous learning | Mentorship | Remote work | Travel up to 30 percentSenior-level Full TimeUnited States R13h ago
-
Enterprise Applied AI Solution Specialist USD 90K-130KAPI Integration | Agent Frameworks | Conversational Design | Data Governance | Function CallingMid-level Full TimeWeston,FL,United States R13h ago
-
Data Engineer I - Salesforce Data 360 USD 87K-123KAPIs | Access Controls | Data Governance | Data Lineage | Data Modeling401k match | Career growth and development programs | Comprehensive onboarding | Conference participation | Employee resource groupsMid-level Full TimeMorrisville, North Carolina (NC), United States R14h ago
-
Applied AI Engineer USD 171K-257KAMD | Agentic Systems | Alignment Frameworks | CRM | ContainerizationCompany-sponsored team events | Flexible time off | Travel opportunities | Wellness resourcesSenior-level Full TimeRemote, United States R15h ago
-
Senior Database Engineer USD 140K-160KAWS | AWS RDS | Amazon Aurora | Amazon Redshift | CI/CD401k | Employee assistance program | HSA | Life insurance | Long-term disabilitySenior-level ContractFoster City, California, USA R15h ago
-
Computational Scientist USD 85K-139KAlgorithm Optimization | Backend Development | Computational modeling | Computer Vision | Data AnalysisRemote Hybrid eligibleMid-level Full TimeUniversity R17h ago
-
Automated testing | Board Farm | Board Farm Automation | C++ | Continuous integration401k | Employee assistance program | HSA | Life insurance | Long-term disabilityMid-level ContractChicago, Illinois, USA R19h ago
-
Senior AI Engineer – AI Platform USD 140K-160KAWS | Agentic AI | Alerting | Alignment | Distributed SystemsDaily meal specials | Dry cleaning | Free shuttle transportation | Gourmet cafeteria | Group fitness classesSenior-level Full TimeEnglewood Cliffs, NEW JERSEY, United States R19h ago
-
GenAI Engineer, Associate USD 85K-110KAPI | AWS | Agentic Coding Tools | Agentic Workflows | Agentic codingRemote workMid-level Full TimeChicago - 550 Van Buren, United … R1d ago
-
Early Career Consult Program - Associate AI Engineer USD 129K-279KAWS | Azure | CI/CD | Cloud Computing | Data AnalysisAccess to learning opportunities | Be Well programs | Coaching | Continuous feedback | Employee referral programMid-level Full TimeUSDALFRI - Dallas Frisco AI HUB … R1d ago
-
AI | AWS | Automation | Azure | Data AnalysisOnsite 3 Days Per Week Requirement | Remote workMid-level Full TimeSouth Carolina Work at Home, United … R1d ago
-
3D Reconstruction | 6DoF Pose | C++ | Camera Calibration | Computer VisionSenior-level Full TimeRemote, United States, United States R1d ago
-
Data Engineering Tech Lead USD 10KAPI Development | Apache Spark | Automated unit testing | Azure | Azure DataSenior-level Full TimeWORK AT HOME - OHIO, United … R1d ago
-
Senior Cloud Data Engineer - Remote USD 91K-163KAgile | Airflow | Apache Spark | Azure | Azure Data4 days per week office requirement in specific locations | Development opportunities | Flexible remote work | Performance recognition | Remote work within U.S.Senior-level Full TimeEden Prairie, Minnesota, UNITED STATES; REMOTE R1d ago
-
Data Engineer (Hybrid) USD 86K-165KAWS | AWS Glue | Access Control | Agile | Amazon EMR401k matching | Adoption Assistance | Child and elder care services | Disability coverage | Employee assistance planMid-level Full TimeUS-IA-CEDAR RAPIDS-193 ~ 1120 Collins Rd … R1d ago
-
Analyst - Data Engineering USD 95KAPI Integration | AWS | Apache Spark | Automated testing | AzureHybrid work schedule | Inclusive and diverse work culture | Professional development opportunities | Supportive leadershipMid-level Full TimeLouisville - West Market, United States R1d ago
-
Engineer II - AI/ML Engineering USD 72K-129KAWS | Bedrock | Cloud Compute Cost Optimization | Cloud compute | Cloud platformAdoption Assistance | Health insurance | Hybrid work flexibility | Identity theft protection | Matched 401kMid-level Full Time600 Hidden Ridge, Irving, TX (TX0326), … R1d ago
-
Anomaly Detection | Batch Processing | Cause analysis | Compute efficiency | Cost Optimization401k employer match | Flexible time off | Health plans | Mental health support | Paid time offSenior-level Full TimeUSA - Remote, United States R1d ago
-
EDW Developer Senior (100% Remote) USD 91K-163KAI Engineering | Azure | Cause analysis | Copilot Studio | Data Governance100 percent remote work | Full-time employmentSenior-level Full TimeBroadmoor Campus, United States R1d ago
-
Principal Engineer-AI/ML Engineering USD 189K-231KCloud Computing | Data Analysis | Data Mining | Feature Engineering | Machine LearningHybrid work flexibility | Work from homeSenior-level Full TimeOne Verizon Way, Basking Ridge, NJ … R1d ago