Staff Compiler Engineer - PyTorch + Kernel DSLPLATE
San Jose, California, United States
USD 163K-253K Senior-level Full Time
Tasks
- Adapt torch compile to backend
- Build kernel DSLs
- Contribute upstream to PyTorch and Triton
- Define fusion and specialization rules
- Design tensor placement and migration
- Engage in compiler upstream review
- Implement parallelism aware lowering
- Implement placement and scheduling passes
- Lower DSLs to ISA and memory hierarchy
- Lower Inductor IR to hardware
- Optimize kernel autotuning
- Overlap compute with data movement
- Perform fusion tiling and memory planning
- Support tensor pipeline expert and sequence parallelism
Perks/Benefits
- 401k
- Adoption support stipend
- Charitable giving match
- Fertility care stipend
- Flexible work environment
- Holidays
- Medical travel support
- Medical/Dental/Vision insurance
- On demand therapy sessions
- Onsite cafe
- Onsite gym
- Paid time off
- Sick leave
- Virtual fitness classes
- Virtual vet care
Skills/Tech-stack
Autotuning | Collective Primitives | Cost Based Compilation | Custom ISA | Cutlass | Distributed Systems | Expert parallelism | Fusion | GPU Kernel | HPC | Helion | IREE | Inductor | Intermediate representation | Kernel DSL | LLVM | MLIR | Memory planning | NUMA | Performance Modeling | Pipeline parallelism | Placement | PyTorch | Python | Scheduling | Sequence parallelism | TVM | Tensor Parallelism | Tiling | Torch compile | Triton | XLA
Education
Related jobs
-
Featured Feat. Principal Knowledge & Data Architect USD 174K-284KAWS Neptune | Canonicalization | Chunking | Cypher | DBTBenefits including health and wellness programs | Health, wellness, and retirement plansSenior-level Full TimeHeadquarters - Chevy Chase, MD R6d ago
-
Senior Staff Engineer, GDC AI Inference Platform USD 262K-365KAgent coordination | C++ | Cloud infrastructure | Containerization | Data StructuresBonus | Equity | Health insurance | Paid time off | Retirement plansSenior-level Full TimeSunnyvale, CA, USA; Kirkland, WA, USA5h ago
-
Sr AI Engineer USD 115K-142KAgent systems | Agentic Workflows | CI/CD | Citation Grounding | Cloud ComputingSenior-level Full TimeAda, MI, United States, 493559h ago
-
PIIM Analytics Engineer USD 86K-132KAPIs | Access Control | Azure Databricks | Data Architecture | Data GovernanceDental insurance | Medical insurance | Optional Disability Insurance | Paid time off | Supplemental insuranceMid-level Full TimeClay, NY, US11h ago
-
PIIM Analytics Engineer USD 105K-160KAI/ML | APIs | Access Control | Azure Databricks | Code AutomationFull time office or project telework | Relocation assistanceMid-level Full TimeHouston, TX, US11h ago
-
Senior-level Full TimeSeattle, Washington, United States; USA - …12h ago
-
Staff Robotics Engineer, Maritime USD 254K-336KC++ | Collision Avoidance | Computer Vision | Computer Vision Algorithms | Machine LearningHealth benefits | Mentorship | Recovery BenefitsSenior-level Full TimeQuincy, Massachusetts, United States13h ago
-
Senior Robotics Engineer, Maritime USD 220K-292KC++ | Cameras | Computer Vision | Gazebo | LidarHealth insurance | Mentorship | Travel opportunitiesSenior-level Full TimeQuincy, Massachusetts, United States13h ago
-
Machine Learning Operations Engineer USD 140K-240KAWS | Alerting | Cloud GPU | Compression | Data Quality401k | Dental insurance | Health insurance | Office food and beverages | Stock option planMid-level Full TimeRWC HQ14h ago
-
Distributed Systems | Fault Tolerance | GPU Computing | JAX | Low LatencyBonus program | Company benefits program | Equity incentive planSenior-level Full TimeMountain View, California15h ago
-
Sr. Validation & Tools Engineer - Embedded Firmware USD 160K-190KAppium | Bug tracking | C plus plus | C# | CAN401k match | Dental insurance | FSA | Family building benefits | Fertility benefitsSenior-level Full TimePalo Alto15h ago
-
RFML Engineer, EW USD 166K-220K5G | C++ | Channel equalization | Channel estimation | Deep learningEmployee benefits package | Travel | U.S. Secret security clearance supportMid-level Full TimeCosta Mesa, California, United States16h ago
-
Staff ML Engineer, StockStory USD 140K-160KCNN | Deep learning | Embeddings | Human-in-the-loop | Information ExtractionDaily group classes | Daily specials | Dry cleaning | Free onsite fitness center | Free shuttle transportationSenior-level Full TimeEnglewood Cliffs, NEW JERSEY, United States17h ago
-
Staff Engineer, Data Analysis (R5290) USD 121K-197KAnomaly Detection | Automated Validation | Cloud Storage | Communications | DashboardsSenior-level Full TimeUnited States17h ago
-
Batch scheduling | Batching | CI/CD | Caching | Capacity PlanningFlexible working hours | Generous vacation | Optional equity donation matching | Parental leaveSenior-level Full TimeSan Francisco, CA18h ago
-
Staff Research Engineer, Discovery Team USD 350K-850KArtificial Intelligence | Cloud deployment | Containerization | Data Pipelines | Distributed SystemsFlexible remote work policy | Flexible working hours | Generous vacation | Parental leave | Visa sponsorship assistanceSenior-level Full TimeSan Francisco, CA18h ago
-
Software Engineer, Systems USD 150K-250KAWS S3 | Apache Iceberg | Apache Parquet | C# | C++401k match | Catered meals | Commuter benefit | Dental insurance | Flexible PTOMid-level Full TimeSan Francisco18h ago
-
Senior AI Solutions Engineer USD 164K-215KAI Observability | Agentic Systems | Airflow | Amazon S3 | Amazon Web Services401k plan | Annual Caltrain Pass | Fastrak reimbursement | Fitness reimbursement | Health, dental, vision insuranceSenior-level Full TimeUS - Remote R18h ago
-
C++ | Deep learning | Distributed Training | ETL Pipelines | GoSenior-level Full TimeMountain View, CALIFORNIA, United States18h ago
-
C++ | Deep learning | Distributed Training | ETL | GoSenior-level Full TimeMountain View, CALIFORNIA, United States18h ago
-
Benchmarking | C++ | CUDA | CUDA programming | CUDALibs401k match | Flexible PTO | Medical/Dental/Vision insurance | Paid parental leave | Tuition reimbursementSenior-level Full TimeSunnyvale, CA / Bellevue, WA18h ago
-
Senior-level Full TimeOakland, CA, United States18h ago
-
Active Learning | Agentic Workflows | Bayesian optimization | Computational Biology | Experimental DesignSenior-level Full TimeSomerville, MA19h ago
-
AI Data Engineer USD 133K-185KAmazon Redshift | BigQuery | CI/CD | Claude Code | Clay401(k) self contribution | BetterUp coaching | Company breaks | Development plan | Flexible paid time offSenior-level Full TimeAustin, TX19h ago
-
AWS | Cassandra | Data Processing | Data Processing Pipelines | Distributed SystemsSenior-level Full TimeSan Francisco Bay Area; Seattle19h ago