1 to 2 of 2 TensorFlow Jobs in Lancashire

PREFERRED QUALIFICATIONS - 2 years experience leading teams that in Machine Learning development including building and training large models, working with Pytroch and/or Tensorflow using large distributed fleets of GPU or other accelerated systems. Amazon is committed to a diverse and inclusive workplace. Amazon is an equal opportunity more »

Compiler team is developing a deep learning compiler stack that takes state of the art LLM and Vision models created in frameworks such as TensorFlow, PyTorch, and JAX, and makes them run performantly on our accelerators. The team is comprised of some of the brightest minds in the engineering … or more of the following programming languages: C++, C, Python PREFERRED QUALIFICATIONS - Masters degree or PhD in computer science or equivalent - Experience optimizing Tensorflow, PyTorch or JAX deep learning models - Experience with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM Amazon is committed to a diverse and inclusive more »