- NVIDIA (Santa Clara, CA)
- …or equivalent technologies + Experience with domain-specific language design and compiler optimizations, in particular sparse compilers (MLIR or TACO) + Excellent ... C++, Python, and CUDA programming skills + Strong collaboration, communication, and documentation habits and ideally experience with working in a globally distributed organization Ways to stand out from the crowd: + Strong understanding of sparse computations,… more
- Amazon (Cupertino, CA)
- …accelerators. You will work closely with teams across AWS Neuron including compiler , training and inference optimization to optimize frameworks for AWS's accelerator ... architectures, and engage closely with the PyTorch and JAX and other ML Framework communities to take advantage of their latest capabilities and improve performance and usability for ML model developers. A successful candidate will have a experience developing… more
- IBM (San Jose, CA)
- …examples of job responsibilities include (a) Develop software optimization in the compiler for AI accelerators, (b) Investigate and prototype AI model optimizations ... for their execution in accelerators, (c) Contribute to the development of cutting-edge demonstrations with emerging AI accelerators **Required technical and professional expertise** * Have strong background on AI accelerators and/or AI accelerator-associated… more
- NVIDIA (Santa Clara, CA)
- …analysis for training/inference workload + Knowledge of Linux device drivers and/or compiler implementation + Knowledge of GPU and/or CPU architecture and general ... computer architecture principles #LI-Hybrid Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 184,000 USD - 287,500 USD for Level 4, and 224,000 USD - 356,500 USD for… more
- Meta (Menlo Park, CA)
- …stack with one of the following core focus areas: AI frameworks, compiler stack, high performance kernel development and acceleration onto next generation of ... their performance 3. Analyze deep learning networks, develop & implement compiler optimization algorithms 4. Collaborating with AI research scientists to accelerate… more
- Amazon (Sunnyvale, CA)
- …Amazon Devices is an inventive research and development company that designs and engineer high-profile devices like the Kindle family of products, Fire Tablets, Fire ... workloads to run efficiently on our accelerator * Collaborate closely with compiler engineers, model developers, hardware architects and product teams to build the… more
- Google (Mountain View, CA)
- …and mobile; the list goes on and is growing every day. As a software engineer , you will work on a specific project critical to Google's needs with opportunities to ... overall performance across key ML workloads (JAX/LLMs), software stack (accelerator software/ compiler ), and hardware (GPU, TPU, other). We interact with product… more