Software Engineer, AI Kernels & Performance Optimization — MTIA Software
Bellevue, WA | Menlo Park, CA | New York, NY
USD 175K-255K (estimate) Senior-level Full Time
Tasks
- Build and extend kernel authoring frameworks templates and libraries
- Deliver and maintain PyTorch operator kernel coverage
- Design implement optimize high performance compute and communication kernels for MTIA accelerators
- Develop performance measurement methodology and technical direction
- Investigate numerics and precision trade offs
- Mentor engineers on accelerator programming and performance methodology
- Partner on hardware software co design with silicon teams
- Profile and root cause performance across the full stack
- Work with compiler runtime framework and product teams to deliver production wins
Perks/Benefits
- N/A
Skills/Tech-stack
Attention kernels | C++ | CUBLAS | CUDA | CUDNN | Code generation | Codegen | Collective communication | Compiler backends | Compiler code generation | CuTe | Cutlass | Distributed Systems | FP8 | FPGA | FlashAttention | HIP | Halide | INT4) | INT8 | KV cache | LLVM | ML Compiler | ML compiler backends | MLIR | Model Parallelism | NCCL | OpenCL | PyTorch | Python | Quantization | RCCL | RTL | Roofline modeling | SYCL | Simulation | TCutlass | TVM | Tensor Parallelism | Triton | XLA
Regions
Countries
States
Related jobs
-
Senior Data Engineer ID71671 USD 155K-170KAWS Kinesis | Apache Airflow | Apache Kafka | Automated testing | Batch ProcessingAnnual learning budget | Flexible hours | Learning budget | Mentorship | Mentorship supportSenior-level Full TimeNew York, United States2h ago
-
Senior Data Engineer ID71671 USD 155K-170KAWS Kinesis | Apache Airflow | Automated testing | CI/CD | DBTCollaborative culture | Flexible hours | Internal tech talks | Learning budget | MentorshipSenior-level Full TimeRichmond, United States2h ago
-
Senior Data Engineer ID71671 USD 155K-170KAWS Kinesis | Apache Airflow | Apache Kafka | Automated testing | CI/CDAnnual learning budget | Flexible hours | Internal tech talks | Local well being programs | MentorshipSenior-level Full TimeMiami, United States2h ago
-
Senior Data Engineer ID71671 USD 155K-170KApache Airflow | Automated testing | CI/CD | Dagster | Data LineageAnnual learning budget | Collaborative culture | Flexible hours | Mentorship | People focused supportSenior-level Full TimeOrlando, United States2h ago
-
Senior Data Engineer ID71671 USD 155K-170KAWS Kinesis | Apache Airflow | Apache Kafka | Automated testing | CI/CDAnnual learning budget | Flexible hours | Mentorship | Performance reviews | Remote workSenior-level Full TimeJacksonville, United States2h ago
-
Senior Data Engineer ID71671 USD 155K-170KAWS | AWS Kinesis | Apache Airflow | Apache Kafka | AzureAnnual learning budget | Flexible hours | Health and wellbeing programs | Learning budget | MentorshipSenior-level Full TimeWest Palm Beach, United States2h ago
-
Senior Data Engineer ID71671 USD 155K-170KApache Airflow | CI/CD | DBT | Dagster | Data LineageFlexible hours | Internal tech talks | Learning budget | Local support programs | MentorshipSenior-level Full TimeDallas, United States2h ago
-
Senior Data Engineer ID71671 USD 155K-170KAWS Kinesis | Apache Airflow | Apache Kafka | Automated testing | Batch ProcessingAnnual learning budget | Collaborative culture | Flexible hours | Mentorship | People focused supportSenior-level Full TimeAtlanta, United States2h ago
-
Senior Data Engineer ID71671 USD 155K-170KAWS Kinesis | Amazon Web Services | Apache Airflow | Apache Kafka | Automated testingAnnual learning budget | Flexible hours | Mentorship | Remote work | TechtalksSenior-level Full TimeSan Francisco, United States2h ago
-
Senior Data Engineer ID71671 USD 155K-170KAWS Kinesis | Apache Airflow | Apache Kafka | Automated testing | Batch ProcessingFlexible hours | Internal tech talks | Learning budget | Mentorship | People focused supportSenior-level Full TimeChicago, United States2h ago
-
Senior Data Engineer ID71671 USD 155K-170KAmazon Kinesis | Apache Airflow | Apache Kafka | CI/CD | DBTFlexible hours | Learning budget | Mentorship | Remote work | Well-being programsSenior-level Full TimeTallahassee, United States2h ago
-
Senior Data Engineer ID71671 USD 155K-170KAWS Kinesis | Apache Airflow | Automated testing | CI/CD | DBTFlexible hours | Learning budget | Mentorship | Performance reviews | Remote workSenior-level Full TimeBoston, United States2h ago
-
Senior Data Engineer ID71671 USD 155K-170KAWS Kinesis | Apache Airflow | Apache Kafka | CI/CD | DBTAnnual learning budget | Flexible hours | Mentorship | Performance and compensation reviews | Remote workSenior-level Full TimeIrving, United States2h ago
-
Senior Data Engineer ID71671 USD 155K-170KApache Airflow | Automated testing | Batch Processing | CI/CD | DBTAnnual learning budget | Flexible hours | Mentorship | Performance reviews | Remote workSenior-level Full TimeTexas City, United States2h ago
-
Senior Data Engineer ID71671 USD 155K-170KAWS Kinesis | Apache Airflow | Apache Kafka | Automated testing | CI/CDFlexible hours | Health and wellness programs | Learning budget | Mentorship | Remote workSenior-level Full TimeJersey City, United States2h ago
-
SOFTWARE ENGINEER – Algorithm Development - 7+ yrs of Experience - TS/SCI w/Poly clearance is required- RC - 306/309 A USD 181K-192KAlgorithm Development | C++ | Data queries | Database interfaces | Documentation401k retirement plan | Dental insurance | Floating holidays | Holidays | Life insuranceSenior-level Full TimeColumbia, United States2h ago
-
Data Preparation | Data Visualization | Diagnostic Testing | Feature Engineering | Machine LearningActive Secret clearance or above | Security clearance providedMid-level Full TimeRosslyn, Virginia, United States3h ago
-
Software Engineer - AI Compute Infrastructure USD 148K-300KCI/CD | Deep learning | GPU Acceleration | Kubernetes | LLM InferenceSenior-level Full TimeSeattle, Washington, United States4h ago
-
Tech Lead Software Engineer - AI Compute Infrastructure USD 244K-450KC++ | CI/CD | Containerization | Distributed Systems | GPU AccelerationSenior-level Full TimeSan Jose, California, United States4h ago
-
Senior Software Engineer - AI Compute Infrastructure USD 202K-368KC++ | GPU Acceleration | Go | Kubernetes | Language ModelsSenior-level Full TimeSeattle, Washington, United States4h ago
-
Senior Software Engineer- Metadata Storage USD 218K-387KDeployment Automation | Distributed Systems | Distributed coordination | Distributed locking | Key-Value StorageSenior-level Full TimeSan Jose, California, United States4h ago
-
Senior Research Engineer / Scientist - Storage for LLM USD 207K-368KCUDA | Caching | Data Structures | Distributed Systems | Eviction PolicyConference attendance | Open source contributions | Research resourcesSenior-level Full TimeSeattle, Washington, United States R4h ago
-
LLM AIOps Development Engineer - Data Center Networking USD 207K-368KAPIs | Anomaly Detection | Cause analysis | Closed Loop | Closed-loop automationExecutive-level Full TimeSeattle, Washington, United States4h ago
-
Research Engineer / Scientist - Storage for LLM USD 153K-300KCUDA | Caching | Distributed Systems | Eviction policies | GPU ComputingConference attendance | Innovation-driven culture | Open source contributions | Research resourcesEntry-level Full TimeSeattle, Washington, United States R4h ago
-
AI Research Scientist, VLM (vision language models) USD 153K-260KComputer Vision | Generative AI | Language Models | Language Processing | Large Language ModelsEntry-level Full TimeMenlo Park, CA5h ago