Helix AI Engineer, Training Performance
Tasks
- Benchmark and iterate on custom kernels
- Build performance monitoring and regression detection tooling
- Co design model architectures and training recipes
- Collaborate on accelerator choice and cluster topology
- Determine parallelism strategy configuration
- Evaluate emerging accelerator architectures
- Extend kernel compilers and iteration tooling
- Improve checkpointing and fault tolerance
- Optimize data loading and preprocessing pipelines
- Optimize distributed training performance
- Write and optimize custom kernels
Perks/Benefits
- N/A
Skills/Tech-stack
Activation checkpointing | C++ | CUDA | Context Parallelism | Data parallelism | Expert parallelism | FSDP | InfiniBand RoCE | Infiniband | Mixed Precision | Model Parallelism | NCCL | NVLink | Nsight Compute | Nsight Systems | PyTorch | Python | RDMA | Sequence Packing | Triton
Education
Related jobs
-
Bash | Cloud platform | Data Processing | Docker | GCPAsynchronous work culture | Competitive compensation | Flexible remote work | Friendly work environmentMid-level Full TimeSan José, Costa Rica22h ago
-
CUDA | Computation graphs | Deep learning | Language Models | Large Language ModelsEntry-level Full TimeSan Jose, California, United States23h ago
-
Compression | Data Engineering | Data Pipelines | Distributed Systems | High AvailabilitySenior-level Full TimeSan Jose, California, United States1d ago
-
Data Scientist Graduate (Multimedia) - 2027 Start USD 128K-177KA/B | A/B Testing | B testing | Causal Inference | Data MiningEntry-level Full TimeSan Jose, California, United States1d ago
-
DPO | DeepSpeed | Distributed Training | Evaluation Pipelines | Human preference learningEntry-level Full TimeSan Jose, California, United States1d ago
-
Senior Data Engineer (Multiple Positions) USD 220K-387KApache Flink | Apache Spark | Architecture Design | Data Governance | Data ModelingSenior-level Full TimeSan Jose, California, United States1d ago
-
C++ | Clustering | Computer Vision | Content deduplication | Data EngineeringEntry-level Full TimeSan Jose, California, United States1d ago
-
Adversarial Robustness | Apache Hadoop | Apache Hive | Apache Spark | C++Senior-level Full TimeSan Jose, California, United States1d ago
-
Bottleneck identification | C# | C++ | Code review | DebuggingSenior-level Full TimeSan Jose, CA, USA1d ago
-
Senior Software Engineer, Embedded Systems/Firmware, XR USD 174K-252KAlgorithms | C# | C++ | Code review | Data StructuresSenior-level Full TimeSan Jose, CA, USA; Kirkland, WA, …1d ago
-
AI Storage Solutions Expert USD 224K-356KAIOps | Access Control | Anomaly Detection | Block Device | Block device managementSenior-level Full TimeSan Jose, CA / Austin, TX-%LABEL … R1d ago
-
Senior Software Engineer - Storage Control Plane USD 266K-395KBlock Storage | C# | C++ | CI/CD | CephSenior-level Full TimeSan Francisco Office (Fremont St) R1d ago
-
AWQ | C++ | CUDA | FP8 | GPU ArchitectureSenior-level Full TimeSan Jose, CA1d ago
-
[8GT] Senior Data Engineer USD 120K-155KAWS | BigQuery | CDK | Domo | DynamoDBAgile environment | Great Place to Work certification | Mentoring opportunitiesSenior-level Full TimeSan José, San José Province, Costa …1d 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 R1d ago
-
Mid Machine Learning Engineer USD 145K-190KAWS SageMaker | Airflow | Azure Machine Learning | DBT | GitRemote workSenior-level Full TimeBolivia; Colombia; Costa Rica; Peru R1d ago
-
Data Processing | Deep learning | Distributed Computing | Feature Engineering | MLOpsMid-level Full TimeSan Jose, California, United States2d ago
-
A/B | A/B Testing | Apache Flink | Apache Spark | B testingEntry-level Full TimeSan Jose, California, United States2d ago
-
Algorithms | C++ | CUDA | Computer Systems | Continuous batchingCareer developmentEntry-level Full TimeSan Jose, California, United States2d ago
-
Algorithms | C++ | CUDA | Computer Systems | Continuous batchingEntry-level Full TimeSan Jose, California, United States2d ago
-
Machine Learning Engineer Graduate (E-Commerce Recommendation Foundation)- 2027 Start (PhD) USD 129K-256KDeep learning | Generative Recommendation | Language Models | Large Language Models | Machine LearningEntry-level Full TimeSan Jose, California, United States2d ago
-
Machine Learning Engineer Graduate (E-Commerce Recommendation Foundation) - 2027 Start USD 118K-187KDeep learning | Generative Recommendation | Information Retrieval | Language Models | Language ProcessingEntry-level Full TimeSan Jose, California, United States2d ago
-
Agentic Systems | Causal Inference | Contrastive Learning | Debiasing | Deep learningOnboarding supportEntry-level Full TimeSan Jose, California, United States2d ago
-
Data Scientist – AI & Predictive Analytics USD 106K-142KData analytics | Hadoop | Hive | Machine Learning | NoSQL401k matching | Employee assistance program | Flexible spending account | Health insurance | Health savings accountEntry-level Full TimeSan Jose, CA, United States2d ago
-
Agent Evaluation & Evolution Machine Learning Engineer Intern (AML-Ark-US) - 2027 Summer USD 162K-316KAutomation | Benchmarking | Data Analysis | Language Models | Language ProcessingCareer development workshops | Hands-on projects | Industry exposure | Social eventsEntry-level InternshipSan Jose, California, United States2d ago