Software Engineer, SystemML - Scaling / Performance
Menlo Park, CA | New York, NY
USD 152K-287K (estimate) Mid-level Full Time
Tasks
- Build performance benchmarks and tuners for NCCL and PyTorch
- Develop AI infrastructure and training components for large scale models
- Enable reliable scalable distributed ML training
- Improve distributed GPU reliability and performance
Perks/Benefits
- N/A
Skills/Tech-stack
CUDA | Data-parallel | Distributed Data Parallel | Distributed data | Fully Sharded Data Parallel | HPC | Infiniband | NCCL | Pipeline Parallel | PyTorch | RoCE | Tensor Parallel | TensorFlow
Education
Roles
Regions
Countries
States
Related jobs
-
LLM & Agentic AI Machine Learning Software Engineer, TikTok - Content Ecology Local Services USD 145K-355KAI Agents | Causal Inference | Computer Vision | Conversion Rate | Conversion rate optimizationSenior-level Full TimeSan Jose, California, United States1h ago
-
Senior Research Scientist/Software Engineer, LLM/Agent Platform (TikTok-Content Ecology AI Innovation & Platform) USD 160K-230KCUDA | Deep learning | DeepSpeed ZeRO | Distributed Systems | Evaluation FrameworksSenior-level Full TimeSan Jose, California, United States1h ago
-
Tax Data Systems and Process Analyst USD 116K-166KAPI Integration | Data Warehouse | GRPC | Java | JavaScriptMid-level Full TimeChicago, IL, USA2h ago
-
Research Engineer, Gemini Code Post-training, DeepMind USD 174K-252KBenchmarking | Data Curation | Data Generation | Distributed Training | Evaluation benchmarksMid-level Full TimeMountain View, CA, USA2h ago
-
Senior Machine Learning Engineer USD 149K-224KAIOps | AWS | Apache Spark | Cloud platform | Data Analysis401k match | Career growth opportunities | Dental insurance | Employee resource groups | Flexible paid time offSenior-level Full TimeRemote, United States4h ago
-
Mid-Level Data Engineer USD 90K-115KAPI Development | Azure | Azure Data | Azure Data Factory | Azure Data LakePublic trust clearance support | Remote workMid-level Full TimeWork from home, VA, United States4h ago
-
AI Model Developer USD 125K-150KAWS | Angular | Azure | CI/CD | D3.jsHybrid remote onsite work | Public trust clearance support | Secret clearance supportMid-level Full TimeFAIRFAX, VA, United States4h ago
-
Lead AI Engineer-Machine Learning USD 100K-115KAWS | Agile | Azure | CI/CD | Computer Vision401k match | Dental insurance | Disability insurance | Growth opportunities | HolidaysSenior-level Full TimeParsippany, NJ, United States7h ago
-
AI Engineer USD 73K-170KAWS Bedrock | Amazon Web Services | Azure | Azure OpenAI | Azure OpenAI ServicePaid Holidays | Paid time off | Rewards and Recognition Program | Work-life balanceMid-level Full TimeUnited States8h ago
-
Data Engineer Stf - E4 USD 131K-243KAccess Control | Apache NiFi | Apache Spark | Data Modeling | Data Processing401k match | Dental insurance | Education assistance | Employee assistance program | Flexible spending accountsSenior-level Full TimeFort Worth, TX, US8h ago
-
Biometrics | C++ | Computer Vision | Machine Learning | Multimodal LearningOffice three days per week | Relocation assistanceMid-level Full TimeSan Francisco R13h ago
-
AI in Residence, Computational Protein Design USD 120K-180KData Curation | Deep learning | Experiment design | Interpretability | Language ProcessingCohort support | Mentorship | Structured feedbackEntry-level Full TimeSeattle, Washington, United States14h ago
-
Lead AI Engineer USD 180K-280KAWS SageMaker | Azure ML | CI/CD | ColBERT | Docker401k match | Bonus | Childcare benefits | Dental insurance | Disability insuranceSenior-level Full TimeQuincy, MA, United States15h ago
-
Senior Machine Learning Engineer II USD 125K-180KCI/CD | Computer Vision | Containerization | Deep learning | Image ProcessingSenior-level Full TimeTucson, AZ16h ago
-
Senior Machine Learning Engineer II USD 153K-229KCI/CD | Computer Vision | Containerization | Deep learning | Image ProcessingSenior-level Full TimeChantilly, VA16h ago
-
Senior Machine Learning Engineer II USD 153K-229KCI/CD | Computer Vision | Containerization | Deep learning | Image Processing401k match | Complimentary snacks and beverages | Dental coverage | Disability insurance | Discounted gym membershipsSenior-level Full TimeLong Beach, CA16h ago
-
Senior Machine Learning Engineer I USD 125K-180KCI/CD | Computer Vision | Containerization | Deep learning | Image ProcessingSenior-level Full TimeTucson, AZ16h ago
-
Senior Machine Learning Engineer I USD 125K-187KCI/CD | Computer Vision | Containerization | Deep learning | Feature EngineeringSenior-level Full TimeChantilly, VA16h ago
-
Senior Machine Learning Engineer I USD 125K-187KCI/CD | Computer Vision | Containerization | Deep learning | Image Processing401k match | Complimentary snacks and beverages | Disability insurance | Discounted gym membership | Employee stock purchase programSenior-level Full TimeLong Beach, CA16h ago
-
Sr. Machine Learning Research Engineer, Siri Speech USD 166K-296KAlgorithm Optimization | Context Tracking | Deep learning | JAX | Language ProcessingSenior-level Full TimeCupertino17h ago
-
Robotics Engineer II USD 130K-155KActive Learning | C# | C++ | CUDA | CameraDependent Care Account | Disability coverage | Flexible spending account | Flexible vacation | Health savings accountSenior-level Full TimeAnn Arbor, MI - Onsite17h ago
-
Machine Learning Engineer USD 162K-170KComputer Vision | Data Pipelines | Deep learning | Experiment design | Machine Learning401k plan | Dental insurance | Flexible work hours | Long-term disability | Medical insuranceMid-level Full TimeRemote, United States R18h ago
-
Machine Learning Research Engineer, ASE Search USD 174K-264KA/B | A/B Testing | Apache Spark | B testing | Fine TuningMid-level Full TimeSeattle18h ago
-
Senior Machine Learning Engineer, Digital Twin Platform USD 240K-253KAWS | Azure | Computer Vision | Data Pipelines | GCPFlex First work policy | Remote workSenior-level Full TimeUnited States - Remote R18h ago
-
Staff Machine Learning Engineer USD 176K-308KA2A | Asyncio | Embeddings | FastAPI | Java401k match | ESPP | Family leave | Flexible spending accounts | Flexible time offSenior-level Full TimeSanta Clara, California, United States R20h ago