Software Engineer, SystemML - Scaling / Performance
Menlo Park, CA | New York, NY
USD 152K-287K (estimate) Mid-level Full Time
Tasks
- Build performance tuners for distributed training
- Develop AI framework and trainer components
- Enable scalable distributed machine learning training
- Improve distributed GPU communication reliability and performance
- Optimize multi GPU and multi node communication with NCCL
Perks/Benefits
- N/A
Skills/Tech-stack
CUDA | Data-parallel | Distributed Data Parallel | Distributed data | Fully Sharded Data Parallel | HPC | Infiniband | Machine Learning | NCCL | Pipeline Parallel | PyTorch | RoCE | Tensor Parallel | TensorFlow
Education
Roles
Regions
Countries
States
Related jobs
-
Featured Feat. Associate Director, Data Labs USD 167K-167KAWS | Cloud Computing | Compute Infrastructure | Data Analysis | LLM GovernanceConference speaking opportunities | Hybrid work schedule | Media appearancesSenior-level Full TimeWashington, District of Columbia, 20004, United … R27d ago
-
Strategy & AI Consulting USD 129K-172KAWS | Artificial Intelligence | Azure | Cloud Computing | Cloud platformMid-level Full TimeAtlanta, United States3h ago
-
Process and Analytics Engineer USD 100K-130KAdvanced Pattern Recognition | Agile | Anomaly Detection | Asset Framework | Cause analysisAdoption Assistance | Dental insurance | Disability insurance | Educational assistance | Flexible spending accountMid-level Full TimeWichita, Kansas3h ago
-
Agent Development Kit | Agent systems | Angular | Artificial Intelligence | ClassificationW2 employmentSenior-level Full TimeDearborn, United States3h ago
-
AI Agents | AI Governance | Artificial Intelligence | Enterprise Search | Machine LearningEntry-level ContractAustin, United States4h ago
-
Research Scientist, AI Networking (PhD) USD 148K-287KCUDA | Data-parallel | Deep learning | Distributed Data Parallel | Distributed dataEntry-level Full TimeMenlo Park, CA6h ago
-
Software Engineer, Systems ML (Technical Leadership) USD 160K-210KBias Mitigation | C++ | CUDA | Co-design | Distributed SystemsSenior-level Full TimeBellevue, WA6h ago
-
Senior Staff Software Engineer, AI/ML USD 262K-365KAccelerated inference | Android | Android on device ML | Artificial Intelligence | C++Benefits | Bonus | EquitySenior-level Full TimeSunnyvale, CA, USA6h ago
-
Smart NIC Software Engineer, Embedded USD 147K-211KAI machine learning | C# | C++ | Data center | Data center networkingBonus target | Equity | Health insurance | Paid time off | Retirement benefitsMid-level Full TimeDurham, NC, USA; Raleigh, NC, USA6h ago
-
Computer Vision | Data Processing | Data Structures | Data structures algorithms | Distributed Computing401k match | Health insurance | Paid time offSenior-level Full TimeMountain View, CA, USA6h ago
-
Staff Software Developer, AI/ML, Safety and Security USD 207K-301KComputer Vision | Data Processing | Debugging | Fine Tuning | Language ModelsSenior-level Full TimeWaterloo, ON, Canada; New York, NY, …6h ago
-
Data Engineer, GCS Data Science USD 106K-151KBig Data | C plus plus | Data Documentation | Data Modeling | Data ValidationEntry-level Full TimeNew York, NY, USA; Mountain View, …6h ago
-
Agent Engine | Agentic Development Kit | Agentic development | Cloud platform | Computer VisionSenior-level Full TimeSunnyvale, CA, USA6h ago
-
Staff Software Engineer, Vertex GenAI Evaluation Service USD 207K-301KCloud infrastructure | Computer Vision | Data Processing | Debugging | Distributed SystemsSenior-level Full TimeSunnyvale, CA, USA6h ago
-
Senior Software Engineer, AI/ML GenAI, Google Cloud USD 174K-253KC++ | Computer Vision | Data Processing | Data Structures | Data Structures and AlgorithmsSenior-level Full TimeSunnyvale, CA, USA6h ago
-
Cloud platform | Computer Vision | Data Processing | Debugging | Distributed ComputingSenior-level Full TimeSunnyvale, CA, USA; New York, NY, …6h ago
-
AI Engineer USD 113K-180KAngular | Apache Kafka | CI/CD | Context Management | DevOps401k matching | Charitable matching | Community outreach groups | Flexible paid time off | Life insuranceMid-level Full TimeLeawood, KS, US R12h ago
-
Senior Data Science Engineer USD 116K-198KApache Airflow | CI/CD | DBT | Databricks | Databricks AutoloaderFlexible work optionsSenior-level Full TimeRemote (United States) R17h ago
-
ML Data - Research Scientist USD 143K-264KAudio signal processing | Computer Vision | Data Annotation | Experiment design | Inter-rater reliabilityMid-level Full TimeCupertino18h ago
-
AWQ | Benchmarking | CUDA | CUDA kernels | Chunked prefill401k match | Career growth | Collaborative culture | Disability insurance | FlexibilitySenior-level Full TimePalo Alto, California, United States18h ago
-
CUDA | Checkpointing | DTensor | Data parallelism | DeepSpeed401k plan | Career growth and learning opportunities | Collaborative culture | Disability insurance | Flexibility and ownershipSenior-level Full TimePalo Alto, California, United States18h ago
-
APIs | Behavior Trees | C++ | Dashboards | Data PipelinesSenior-level Full TimeIrvine, CA18h ago
-
Machine Learning Engineer, Local Search & Marketplace USD 190K-300KA/B | A/B Testing | Apache Spark | B testing | C++Senior-level Full TimeMountain View, California, United States18h ago
-
Machine Learning Engineer, Local Search & Marketplace USD 190K-300KA/B | A/B Testing | Algorithms | B testing | C++Senior-level Full TimeMountain View, California, United States18h ago
-
Member of Technical Staff, Applied AI Research USD 203K-299KAgentic technology | Apache Cassandra | Apache Kafka | Caching | Capacity PlanningSenior-level Full TimeSan Francisco, CA; Sunnyvale, CA19h ago