Helix AI Engineer, Training Performance
Tasks
- Build and iterate on custom kernel generation systems
- Build performance monitoring dashboards and tooling
- Co design model architectures and training recipes for scale
- Collaborate on accelerator choice and cluster topology
- Evaluate emerging accelerator architectures and run proof of concept ports
- Explore model parallelism configurations for optimal scaling
- Extend kernel compilers and add custom accelerator support
- Improve checkpointing fault tolerance and elastic restart
- Optimize data loading and preprocessing pipelines
- Optimize training performance for large scale models
- Write and optimize custom GPU kernels
Perks/Benefits
- N/A
Skills/Tech-stack
Activation checkpointing | C++ | CUDA | FSDP | Infiniband | Mixed Precision | NCCL | NVLink | PyTorch | Python | RDMA | Sequence Packing | Torch Profiler | Triton
Education
Related jobs
-
AC DC Sizing | AC/DC | Arc Flash | Artificial Intelligence | AutomationMultiple project extensions | Paid on time | Remote work optionSenior-level Full TimeJuno Beach, FL R19h ago
-
AI Agents | API Development | Azure | CI/CD | LLMOpsCollaborative cross-functional team | Fully remoteMid-level Full TimeCanada, Ontario; Toronto, None; Toronto, Ontario; … R19h ago
-
API Integration | AWS Bedrock | Auditing | Authentication | AuthorizationMentoring | MentorshipMid-level Full TimeMenlo Park, CA, United States20h ago
-
AI Solution Architect USD 38KArtificial Intelligence | Deep learning | Language Processing | MLOps | Machine LearningSenior-level Full TimeMorrisville, North Carolina, États-Unis d’Amérique22h ago
-
Analytics and Insights Engineer III USD 131K-218KApache Spark | Data Governance | Data Modeling | Data Pipelines | Data QualitySenior-level Full TimeRichmond, Virginia, United States22h ago
-
3D Rendering | A/B | A/B Testing | Agentic AI | AlgorithmsSenior-level Full TimeSan Jose, California, United States23h ago
-
C++ | Computer Vision | Data Processing | Deep learning | Domain AdaptationEntry-level Full TimeSan Jose, California, United States23h ago
-
Staff Software Engineer, AI and Infrastructure USD 207K-300KC++ | Cluster management | Distributed Systems | Kernel | LinuxSenior-level Full TimeSunnyvale, CA, USA1d ago
-
Senior Software Engineer, Google Cloud Dataproc USD 174K-252KApache Hive | Apache Spark | C++ | Data Warehouse | Data Warehouse InternalsSenior-level Full TimeSunnyvale, CA, USA1d ago
-
Lead Firmware Engineer, Robotics USD 207K-300KBrushless DC | C# | C++ | Closed Loop | Closed loop controlSenior-level Full TimeMountain View, CA, USA1d ago
-
Agentic Orchestration | C++ | Data Onboarding | Data Pipeline Engineering | Data pipelineSenior-level Full TimeSan Jose, CA, USA1d ago
-
Software Engineer III, Infrastructure, Infra Spanner USD 147K-210KC++ | Concurrency | Data Corruption | Data corruption diagnosis | Database performanceSenior-level Full TimeSunnyvale, CA, USA1d ago
-
Senior Software Engineer, AI/ML, Core USD 174K-252KC++ | Data Processing | Data Structures | Data Structures and Algorithms | DebuggingSenior-level Full TimeMountain View, CA, USA1d ago
-
Staff Software Engineer, Search Local Ads Data Platform USD 207K-300KC plus plus | CSS | Data Storage | Data Structures | Data Structures and AlgorithmsSenior-level Full TimeMountain View, CA, USA1d ago
-
Embedded Hardware Systems Engineer USD 130K-150KAltium Designer | BLE | Bluetooth Low Energy | Bring-up | C++401k matching | Corporate discounts | Dental insurance | Disability insurance | Employee assistance programMid-level Full TimeSan Jose, CA, US R1d ago
-
Analytics Data Engineer USD 94K-130KAzure Data | Azure Data Factory | CI/CD | Data Analysis | Data FactoryRemote workMid-level Full TimeFlagstaff, AZ, US1d ago
-
AI Application & Cloud Operations Engineer (Mid level) USD 108K-117KAWS | AWS Bedrock | AWS CloudTrail | AWS CloudWatch | AWS PrivateLinkMid-level Contract Full TimePlano, TX, United States1d ago
-
Benchmarking | C plus plus | CUDA | Computational Engineering | Distributed Computing401k match | Generous paid time off | Medical, dental & vision coverage | Stock options | WFH Hardware ReimbursementSenior-level Full TimeSunnyvale, CA R1d ago
-
Software Engineer, Applied AI USD 173K-250KABAC | Agentic Frameworks | Apache Airflow | Audit trails | BIMDental insurance | Generous PTO | Health insurance | Retirement or Pension Plan | Vision insuranceMid-level Full TimeSan Francisco, CA1d ago
-
AI Research Scientist - GenAI USD 165K-185KC plus plus | Deep learning | GenAI | Inference Optimization | Language ModelsMid-level Full TimePittsburgh, PA, United States1d ago
-
Staff Data Engineer USD 147K-184KAmazon Kinesis | Amazon Web Services | Apache Airflow | Apache Iceberg | Apache SparkDental insurance coverage | Health insurance coverage | Paid parental leave | Paid vacation | Relocation assistanceSenior-level Full TimeBerkeley, CA1d ago
-
Associate, Client Intelligence (Data Scientist) USD 98K-132KAWS | Apache Spark | Azure | Big Data | Cloud platformEmployee resource groups | Matching gift program | Professional development | Sustainability and Climate Change Initiatives | Volunteerism programMid-level Full TimeNew York, NY, US, 100011d ago
-
Apple Neural Engine | C++ | CPU | Compilers | Computer ArchitectureMid-level Full TimeSeattle1d ago
-
On-Device ML Infrastructure Engineer (CoreML Runtime), Graphics, Games and Machine Learning USD 170K-296KC++ | Core ML | GPU Programming | JAX | LLVMMid-level Full TimeCupertino1d ago
-
Apple Neural Engine | C plus plus | CPU performance | Compilers | Computer ArchitectureMid-level Full TimeCupertino1d ago