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
-
Software Engineer III - Senior Java Spark Developer USD 113K-188KAgile | Apache Spark | CI/CD | Concurrency | Distributed SystemsSenior-level Full TimeJersey City, New Jersey, United States2h ago
-
Applied AI Engineer - AI Solutions USD 172K-300KAgentic Workflows | Airflow | Apache Spark | Chroma | CrewAIAnnual travel up to 25% | Employee stock options | Hybrid work | Professional developmentMid-level Full TimeNew York City, NY (Hybrid); Redwood … R10h ago
-
AI Solutions Engineer, Talent Acquisition USD 129K-171KAPIs | Access Control | Agentic Workflows | Audit trails | AuthenticationMid-level Full TimeSeattle, Washington, United States12h ago
-
Backend Engineer - Applied AI USD 175K-240KCRDT | Caching | Circuit Breakers | Database Query | Database Query OptimizationMid-level Full TimeNew York City16h ago
-
Lead Principal Machine Learning Engineer USD 169K-355KAccess Control | Agent systems | AgentOps | Auditability | Autogen401k match | Life insurance | Paid Holidays | Paid parental leave | Paid sick leaveSenior-level Full TimeSan Francisco, CA, United States16h ago
-
2026 Fall Health AI Scholar, Digital Health Algorithms USD 124K-150KAWS SageMaker | Amazon Redshift | Android | Artificial Intelligence | C++Entry-level Full Time665 Clyde Avenue, Mountain View, CA, …16h ago
-
Senior Software Engineer - Infrastructure Storage USD 266K-395KAPI Design | Block Storage | Ceph | Distributed Systems | Fibre Channel401k match | Commuter stipend | Flexible paid time off | Health, dental, vision coverage | Wellness stipendSenior-level Full TimeSan Francisco Office (Fremont St)17h ago
-
Senior-level Full TimeCosta Mesa, California, United States17h ago
-
Sr. Data Engineer (Remote) USD 163K-192KAccess Control | Amazon Web Services | Apache Iceberg | Apache Kafka | Apache Spark401k plan | Dental insurance | Disability insurance | Employee assistance program | FSA/HSASenior-level Full TimeRemote - United States R17h ago
-
Machine Learning Engineer - Simulation Framework USD 160K-234K3D Graphics | C++ | CUDA | Deep learning | Deterministic systemsEntry-level Full TimeFoster City, CA17h ago
-
Reliability Engineer, Supercomputing USD 350K-475KBMC | Container Orchestration | DCGM | Debugging | Firmware ManagementDental benefits | Health benefits | Paid parental leave | Relocation support | Unlimited PTOMid-level Full TimeSan Francisco17h ago
-
AWS | Agent Orchestration | CI/CD | Cloud platform | Databricks401k match | Counseling membership | Employer subsidized medical dental and vision | Flexible time away program | Life insuranceMid-level Full Time-REMOTE, USA- R17h ago
-
Software Engineer III - Big Data & AWS USD 175K-186KAPI Gateway | AWS Glue | AWS Lambda | AWS Step Functions | Amazon APISenior-level Full TimePlano, TX, United States17h ago
-
AI Solutions Engineer, Talent Acquisition USD 129K-171KAPI Integration | APIs | Access Control | Agentic Workflows | Audit trailsMid-level Full TimeBoston, Massachusetts, United States17h ago
-
AI Solutions Engineer, Talent Acquisition USD 129K-171KAPIs | Access Control | Agentic Workflows | Audit Logs | AuthenticationHealth insurance | Paid time off | Wellness programsMid-level Full TimeCosta Mesa, California, United States17h ago
-
AI Engineer, Ecosystem USD 171K-240KAPI Integration | Access Management | Audit Logging | Authentication | AuthorizationHybrid work | Remote work up to 4 weeks per yearMid-level Full TimeSan Francisco, California, United States R17h ago
-
Algorithms | Apache Airflow | Apache Flink | Apache Spark | C++Senior-level Full TimeUS-WA-Bellevue18h ago
-
Forward Deployed AI Solutions Engineer USD 95K-145KAPIs | Agentic Workflows | Audit Logging | Cloud Computing | Command Line401k benefits | Commuter benefits | Employee referral program | Fertility care benefits | Free testingMid-level Full TimeUS Remote R18h ago
-
Staff Analytics Engineer USD 159K-187KClaude | Claude Code | DBT | Data Contracts | Data Modeling401k company match | Accident insurance | Company funded HSA contributions | Critical illness insurance | Health, dental, vision coverageSenior-level Full TimeRemote (United States) R18h ago
-
AI Engineer, Product USD 171K-240KA/B | A/B Testing | API Design | B testing | Data ModelingHybrid work | Remote work up to four weeks per yearMid-level Full TimeSan Francisco, California, United States R18h ago
-
AI Data Engineer USD 80K-100KAWS Bedrock | AWS CDK | Access Management | Anthropic | Authentication tokens401k | Dental insurance | Health insurance | Vision insurance | Wellness benefitsSenior-level Full TimeLos Angeles, CA, US18h ago
-
C++ | Deep learning | Distributed Training | ETL | GoSenior-level Full TimeMountain View, CALIFORNIA, United States18h ago
-
C++ | ETL | Go | Hugging Face | MicroservicesFlexible work arrangements | Learning and development support | Remote work optionsSenior-level Full TimeMountain View, CALIFORNIA, United States18h ago
-
Staff Machine Learning Engineer, Agentic Product USD 137K-241KAgentic AI | Generative AI | Interface Development | Language Models | Language ProcessingProfessional development support | Remote work flexibilitySenior-level Full TimeMountain View, CALIFORNIA, United States18h ago
-
Abstractive summarization | Active Learning | DPO | Data Privacy | Deep learningIn-office work options | Professional development | Remote work flexibilitySenior-level Full TimeMountain View, CALIFORNIA, United States18h ago