Research Scientist, AI Networking (PhD)
Tasks
- Build software benchmarks and performance tuners
- Develop NCCL based distributed training stack
- Enable scalable distributed ML training
- Improve distributed GPU communication reliability and performance
- Integrate and optimize communication for PyTorch
- Perform HPC and parallel computing optimizations
- Work on GenAI and LLM scaling reliability and performance
Perks/Benefits
- N/A
Skills/Tech-stack
CUDA | Data-parallel | Deep learning | Distributed Data Parallel | Distributed data | Distributed machine learning | Fully Sharded Data Parallel | GPU Architecture | High Performance | High-Performance Computing | Infiniband | Language Models | Large Language Models | Machine Learning | NCCL | Parallel Computing | Performance Computing | Pipeline Parallel | PyTorch | RoCE | Tensor Parallel | TensorFlow
Education
Roles
Engineer | Research Scientist | Scientist | Software Engineer
Regions
Countries
States
Cities
Related jobs
-
Data Scientist USD 145K-190KBayesian statistics | Bremsstrahlung | C++ | C++ and Fortran | Data PipelinesFlexible time off | Medical, dental, and vision coverage | Paid parental leave | Paid time off | Relocation assistanceMid-level Full TimeSan Leandro, California6h ago
-
Sr Data Scientist USD 138K-197KAnomaly Detection | Clustering | Language Processing | Logistic Regression | Machine Learning401k matching | Charitable matching program | Dental insurance | Employee assistance program | Employee wellness programsSenior-level Full TimeCT, US7h ago
-
Senior Machine Learning Engineer, Content Platform USD 141K-360KApache Spark | Cloud Computing | Generative AI | Language Models | Large Language ModelsDisability benefits | Equity awards | Financial wellness support | Health insurance | Life insuranceSenior-level Full TimeSan Jose, California R8h ago
-
Staff Machine Learning Engineer, Ad Serving USD 195K-510KA/B | A/B Testing | B testing | Continuous Improvement | Feature EngineeringCommuter benefits | Disability benefits | Financial wellness support | Health insurance | Life insuranceSenior-level Full TimeNew York, New York R8h ago
-
Staff Machine Learning Engineer, Ad Serving USD 195K-435KDistributed Systems | Experimentation | Feature Engineering | Java | Machine LearningDisability benefits | Equity awards | Health insurance | Life insurance | Paid time offSenior-level Full TimeBoston, Massachusetts R8h ago
-
Machine Learning Engineer, Ads USD 165K-230KA/B | A/B Testing | B testing | Bidding | Conversion PredictionDirect impact | Hybrid work model | International team collaboration | Professional growth opportunities | Remote work flexibilityMid-level Full TimeSan Francisco Bay Area, USA R9h ago
-
Senior Computer Vision Engineer USD 191K-292KComputer Vision | Data Engineering | Dataset development | Deep learning | Edge ComputingHealth benefits | Recovery BenefitsSenior-level Full TimeWashington, District of Columbia, United States11h ago
-
Staff/Principal AI Software Engineer - Snowflake CoWork USD 264K-379KAgent systems | Artificial Intelligence | Function Calling | Guardrails | LLM orchestrationSenior-level Full TimeUS-CA-Menlo Park12h ago
-
Applied AI Lead USD 130K-150KAI Agents | AWS | AWS Bedrock | Azure | Azure AI401k match | College savings | Company paid life and AD D insurance | EAP | FSA optionSenior-level Full TimeHouston, TX, United States13h ago
-
Applied AI Lead USD 140K-160KAI Agents | AWS | AWS Bedrock | Anthropic Claude | Automation401k match | Cell phone stipend | College savings | EAP | FSA optionsSenior-level Full TimePhiladelphia, PA, United States13h ago
-
AWS | Cloud infrastructure | Compute | Data Curation | Distributed Systems401k match | Daily lunches and snacks | Health insurance covered | Hybrid work schedule | Learning and development stipendSenior-level Full TimeRedwood City R14h ago
-
Sr. Forward Deployed Engineer (FDE) - Public Sector USD 182K-250KAWS | Apache Spark | Azure | CI/CD | DatabricksSenior-level Full TimeWashington, D.C.14h ago
-
Sr. Forward Deployed Engineer (FDE) - Public Sector USD 182K-250KAI APIs | AWS | Apache Spark | CI/CD | Cloud platformCustomer-facing role | Hybrid work | Secret clearance required | Travel 20 to 25 percentSenior-level Full TimeAlabama; Arkansas; Florida; New York; Tennessee R14h ago
-
AI Research Engineer – Agentic AI USD 165K-180KAgent systems | Evaluation | Experiment design | Language Models | Large Language Models401k matching | Disability insurance | Health insurance | Life insurance | Paid time offMid-level Full TimeSunnyvale, CA, United States14h ago
-
Lab Automation - Robotics Engineer USD 150K-180KAPI Integration | Accelerated inference | Automation | Behavior Trees | C#Entry-level Full TimeSouth San Francisco, California, United States15h ago
-
AI and Computational Geometry Engineer USD 200K-250KABAQUS | ANSYS | BREP | Boolean Operations | CAEL toolsEquity | Health benefits | Paid time offMid-level Full TimeEmeryville, California15h ago
-
Staff Data Scientist – Simulation Motion AI USD 170K-240KCalibration | Closed Loop | Closed loop control | Computer Vision | Coordinate framesSenior-level Full TimeSan Francisco, CA, United States15h ago
-
Sr. Staff Machine Learning Systems Engineer USD 240K-265KAdversarial Testing | Calibration | Data Engineering | Data Ingestion | Data Pipelines401k employer matching | Company holidays | Comprehensive health benefits | Employee stock purchase program | Offsite team retreatsSenior-level Full TimeUS Remote R15h ago
-
CUPED | Causal Graphs | Causal Inference | Contextual Bandits | Counterfactual evaluation401k plan | Basic life insurance | Commuter benefits match | Disability insurance | Family-forming assistanceSenior-level Full TimeSan Francisco, CA; Sunnyvale, CA; Los …15h ago
-
Senior Data Scientist, Marketing USD 145K-200KAirflow | Attribution Modeling | BigQuery | Cohort Analysis | ComposerHealthcare benefits | Paid Holidays | Paid Scopely free days | Pet insurance | Retirement benefitsSenior-level Full TimeUS - San Francisco, United States15h ago
-
Junior AI Engineer (Open to remote) USD 110K-135KAPIs | Agent SDK | Agentic coding | Arize | Google ADK401k | Commuter benefits | Dental insurance | Educational assistance | Medical insuranceExecutive-level Full TimeNew York, United States R15h ago
-
AI Software Engineer II USD 130K-196KAWS | Agent Orchestration | Amazon Redshift | CI/CD | DBTLearning and development stipend | PTO and sick leave | Parental leaveMid-level Full TimeLong Beach, California, United States16h ago
-
Senior-level Full TimeCalifornia; Remote - California; San Francisco, … R16h ago
-
Research Engineer-Data & Analytics USD 80K-122KCloud Computing | Code review | Data Modeling | Data analytics | Distributed ComputingMid-level Full TimeRaleigh, North Carolina, United States16h ago
-
Principal Machine Learning Engineer, Geometric Vision USD 407K-460K3D Computer Vision | 3D representations | Auto-labeling | C++ | CalibrationSenior-level Full TimeSunnyvale R16h ago