Amazon is seeking a software engineer for the Neuron compiler team in Cupertino to design and implement the next‑generation Neuron compiler. You will translate models from PyTorch, TensorFlow, and JAX into efficient code for Inferentia/Trainium chips, collaborating with cross‑domain teams to optimize ML workloads.
The role requires strong OO language skills (C++/Java), system design experience, and a Master’s/PhD in CS preferred.
#J-18808-Ljbffr