- Cisco (San Jose, CA)
- …like pytorch. + Familiarity with **assembly or PTX/SASS** for debugging or optimizing CUDA kernels . + Familiarity with **NVMe storage offloads** , **IOAT/DPDK** ... the next generation of enterprise-grade AI infrastructure. As a principal engineer within our GPU and CUDA Runtime team, you will play a critical role in shaping… more
- Google (Mountain View, CA)
- …GPU/TPUs via ML frameworks (eg JAX, PyTorch) and low-level programming models (eg CUDA , OpenCL) + Experience in leveraging custom kernels and compiler ... Snapshot We are seeking a software engineer to define, drive, and critically contribute to...implement critical components across Model architecture, ML frameworks, custom kernels and platform, to deliver frontier models with maximum… more
- Amazon (San Francisco, CA)
- …review practices A day in the life - Optimize transformer blocks using custom CUDA kernels and TensorRT optimization techniques - Partner with scientists to ... foundation models run at production scale. As a Senior Machine Learning Engineer embedded in our science team, you'll be instrumental in transforming cutting-edge… more
- Amazon (San Francisco, CA)
- …review practices A day in the life - Optimize transformer blocks using custom CUDA kernels and TensorRT optimization techniques - Partner with scientists to ... breakthrough foundation models run at production scale. As a Software Development Engineer embedded in our science team, you'll be instrumental in transforming novel… more
- Microsoft Corporation (Mountain View, CA)
- …the Firmware Center of Excellence, we're looking for a customer-focused, hands-on SW engineer to help us develop a suite of system validation and diagnostic tests ... geographic locations. We are looking for a Principal Firmware Engineer to join the team. Microsoft's mission is to...PVT characterization. + Knowledge of or experience with AI models/ kernels such as GPT, Gemm, etc. + Experience with… more
- Meta (Menlo Park, CA)
- …frameworks like PyTorch, Caffe2, TensorFlow, ONNX, TensorRT 10. OR AI high performance kernels : Experience with CUDA programming, OpenMP / OpenCL programming or ... directly towards PyTorch code and device optimization. **Required Skills:** Software Engineer - Systems ML - PyTorch Responsibilities: 1. Improve PyTorch's state… more
- Microsoft Corporation (Mountain View, CA)
- …in optimizing machine learning models for GPUs, including development of custom CUDA kernels for performance-critical workloads. Software Engineering IC6 - The ... We are looking for an experienced **Principal Software Engineer ** to join the Ads Engineering team and help advance the core capabilities of our Ads serving stack.… more