Staff Software Engineer - GenAI Performance and Kernel
Tasks
- Benchmark kernels
- Build profiling and instrumentation tools
- Collaborate on production rollout and monitoring
- Design core compute kernels
- Detect performance regressions
- Establish kernel abstractions and frameworks
- Implement compute kernels
- Integrate kernel optimizations into ML systems
- Maintain kernel software
- Mentor engineers
- Optimize kernel performance
- Perform root cause analysis on inference bottlenecks
- Review code and set best practices
- Verify kernel correctness
Perks/Benefits
- N/A
Skills/Tech-stack
Assembly | Auto-tuning | CUBLAS | CUDA | CUDNN | Cutlass | Debugging | Error propagation | Fusion | GPU Architecture | GPU memory | GPU memory hierarchy | LLVM IR | Loop transformations | Memory Reuse | Memory hierarchy | Mixed Precision | Numerical Stability | OneDNN | OpenCL | Profiling | Quantization | Software pipelining | Sparsity | Tensor cores | Tensorization | Tiling | Triton | Vectorization | Warp scheduling
Education
Roles
Regions
Countries
States
Related jobs
-
Algorithms | Android | Audio pipelines | Bare Metal | C++Senior-level Full TimeSan Jose, CA, USA; Kirkland, WA, …4h ago
-
Artificial Intelligence | Data Processing | Debugging | Distributed Systems | Fine TuningSenior-level Full TimeMountain View, CA, USA4h ago
-
Artificial Intelligence | Cloud platform | Code review | Data Structures | Data Structures and AlgorithmsSenior-level Full TimeSunnyvale, CA, USA; Kirkland, WA, USA4h ago
-
Senior Software Engineer, AI/ML, Android USD 174K-252KC++ | Data Processing | Data Structures | Data Structures and Algorithms | DebuggingSenior-level Full TimeMountain View, CA, USA4h ago
-
Senior Software Engineer, Embedded, Pixel Graphics USD 174K-252KC# | C++ | Device Drivers | Embedded Linux | GPGPUSenior-level Full TimeMountain View, CA, USA; San Diego, …4h ago
-
Senior Software Engineer, AI/ML GenAI, Google Cloud USD 174K-252KC++ | Computer Vision | Data Processing | Debugging | GenAISenior-level Full TimeSunnyvale, CA, USA; New York, NY, …4h ago
-
Software Engineer III, AI/ML, Google Cloud Platforms USD 147K-210KC++ | Data Processing | Debugging | Language Processing | ML InfrastructureSenior-level Full TimeSunnyvale, CA, USA4h ago
-
AI/Machine Learning Engineer - Python - Loops USD 130K-150KAPI Development | CI/CD | Data Pipelines | Debugging | Embeddings401k | Community volunteering events | Dental insurance | Disability benefits | Flexible paid time offEntry-level Full TimePalo Alto, California, United States17h ago
-
Lab Automation - Robotics Engineer USD 150K-180KAPI Integration | Accelerated inference | Automation | Behavior Trees | C#Entry-level Full TimeSouth San Francisco, California, United States17h ago
-
Senior-level Full TimeUnited States, California, Mountain View; United …18h ago
-
Software Engineer II USD 120K-145KAPIs | Alerting | Amazon Web Services | Apache Spark | Automated testing401k match | ADandD insurance | Accident insurance | Company holidays | Critical illness insuranceMid-level Full TimeKirkland, WA19h ago
-
Machine Learning Engineer, Detection and Tracking USD 160K-210KByteTrack | DETR | Data Augmentation | DeepSORT | Faster R-CNNFlexible paid time off | Hybrid work | Medical insurance | Paid Holidays | Remote workMid-level Full TimeWashington, DC R20h ago
-
Staff Software Engineer - GenAI inference USD 190K-232KA/B | A/B Testing | Attention Mechanism | B testing | CUBLASSenior-level Full TimeSan Francisco, California22h ago
-
Sr. Machine Learning Engineer - Machine Learning USD 178K-213KAPI Design | CI/CD | CUDA | Custom Kernels | Dependency management401k plan | Annual performance bonus | Commuter support | Company-provided laptop | Flexible paid time offSenior-level Full TimeRemote; Watertown, Massachusetts, United States R22h ago
-
Access Control | Audit compliance | Authentication | Authorization | Confluence401k plan | Best in class onboarding | Continuous career development | Dental benefits | Employee stock purchase planSenior-level Full TimeDenver, Colorado, USA; San Francisco, California, … R1d ago
-
Senior Software Engineer, AI/ML, YouTube USD 174K-252KC++ | Data Processing | Data Structures | Data Structures and Algorithms | DebuggingSenior-level Full TimeMountain View, CA, USA1d ago
-
Senior Software Engineer, AI/ML, Gen AI, Geo USD 174K-252KComputer Vision | Data Processing | Data Storage | Data Structures | Data structures algorithmsSenior-level Full TimeMountain View, CA, USA1d ago
-
Senior Software Engineer, Agentic AI, AIR USD 174K-252KComputer Vision | Data Processing | Data Structures | Data structures algorithms | DebuggingSenior-level Full TimeMountain View, CA, USA; New York, …1d ago
-
Cause analysis | Computer Vision | Data Processing | Debugging | Distributed SystemsSenior-level Full TimePittsburgh, PA, USA; Mountain View, CA, …1d ago
-
Algorithms | C# | C++ | Code review | Computer VisionSenior-level Full TimeSan Jose, CA, USA; San Francisco, …1d ago
-
C++ | Computer Vision | Data Processing | Debugging | Distributed ComputingSenior-level Full TimeBoulder, CO, USA1d ago
-
ATA | Algorithms | C# | C++ | Data StructuresSenior-level Full TimeSunnyvale, CA, USA1d ago
-
Staff Software Engineer, Network Health USD 207K-300KAnomaly Detection | Automated remediation | Data Processing | Debugging | Distributed ComputingSenior-level Full TimeSunnyvale, CA, USA1d ago
-
AI machine learning | Algorithms | C++ | Data Analysis | Data StructuresSenior-level Full TimeRaleigh, NC, USA; Durham, NC, USA1d ago
-
Senior Software Engineer, Perception USD 184K-287K2D Object Detection | 3D Object Detection | C++ | CUDA | CameraSenior-level Full TimeAtlanta - Onsite R1d ago