- Google (Sunnyvale, CA)
- …and serving benchmarks that are representative to Google production, industry and Machine Learning (ML) community, use them to identify performance opportunities ... technical field or equivalent practical experience. + Experience in optimizing GPU -accelerated environments with the understanding of large language models (LLMs)… more
- Amazon (Cupertino, CA)
- …Neuron is the complete software stack for the AWS Inferentia and Trainium cloud-scale machine learning accelerators and the Trn1 and Inf1 servers that use them. ... a Software Engineer that will lead the development of machine learning tools to run, optimize, and...new instruction set architectures, which may include CPU, NPU, GPU and other forms of compute. Key job responsibilities… more
- Google (Sunnyvale, CA)
- …either an academic or industry setting. + 2 years of experience with Machine Learning (ML) algorithms and tools (eg, TensorFlow), Artificial Intelligence (AI), ... will aim to make OpenXLA the best place for Machine Learning (ML) acceleration innovation. OpenXLA is...on Central Processing Units (CPU) and Graphics Processing Units ( GPU ). You will work on the stack (the hardware-independent… more
- Amazon (Cupertino, CA)
- …and JAX. Your role will involve working closely with our custom-built Machine Learning accelerators, including Inferentia/Trainium, which represent the forefront ... Key job responsibilities As a Sr. ML Compiler Engineer III on the Neuron Compiler Automated Reasoning Group, you...in compiler development is strongly preferred. A background in Machine Learning and AI accelerators is preferred,… more