Software Engineer, AI Kernels & Performance Optimization — MTIA Software
Bellevue, WA | Menlo Park, CA | New York, NY
USD 175K-275K (estimate) Senior-level Full Time
Tasks
- Build and extend kernel authoring frameworks and libraries
- Deliver PyTorch operator kernel coverage for AI workloads
- Design high performance compute kernels for MTIA accelerators
- Develop software mitigations for hardware limitations
- Investigate numerics and precision trade offs
- Mentor engineers on performance methodology and accelerator programming
- Optimize kernel performance through architectural analysis
- Partner on hardware software co design with silicon teams
- Profile and root cause performance across full stack
- Set technical direction and write design documents
Perks/Benefits
- N/A
Skills/Tech-stack
C++ | CUDA | Collective communication | Computer Architecture | DMA | Distributed Computing | FP8 | FPGA Emulation | HIP | Halide | INT4) | INT8 | LLVM | Low Precision | Low-precision numerics | MLIR | Memory hierarchy | NCCL | Occupancy | OpenCL | Polyhedral scheduling | PyTorch | Python | Quantization | RCCL | ROCm | Roofline model | SYCL | Scheduling | Simulators | Synchronization | TVM | Vectorization | 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 States3h 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 States3h 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 States3h 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 States3h 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 States3h 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 States3h 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 States3h 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 States3h 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 States3h 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 States3h 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 States3h 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 States3h 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 States3h 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 States3h 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 States3h 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 States3h ago
-
Data Preparation | Data Visualization | Diagnostic Testing | Feature Engineering | Machine LearningActive Secret clearance or above | Security clearance providedMid-level Full TimeRosslyn, Virginia, United States4h ago
-
Software Engineer - AI Compute Infrastructure USD 148K-300KCI/CD | Deep learning | GPU Acceleration | Kubernetes | LLM InferenceSenior-level Full TimeSeattle, Washington, United States5h ago
-
Tech Lead Software Engineer - AI Compute Infrastructure USD 244K-450KC++ | CI/CD | Containerization | Distributed Systems | GPU AccelerationSenior-level Full TimeSan Jose, California, United States5h ago
-
Senior Software Engineer - AI Compute Infrastructure USD 202K-368KC++ | GPU Acceleration | Go | Kubernetes | Language ModelsSenior-level Full TimeSeattle, Washington, United States5h 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 R5h 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 States5h 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 R5h 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, CA6h ago
-
Research Engineer - MSL FAIR Foundations USD 141K-208KBenchmarking | Code review | Data Pipelines | Deep learning | Distributed SystemsMid-level Full TimeMenlo Park, CA | Burlingame, CA …6h ago