for Senior ML Compiler Engineers to join the team building the ML backend (compiler, run-time, and debugger) for our next-generation OPTUs that connect PyTorch, Tensorflow, JAX, and MXNet down to our low-level kernel drivers.Your mission will be to create seamless support for a broad ecosystem of large AI models, and ensure we are pushing the limits of … unlock key high-impact technical and business milestones that drive the success of Flux. Architect & Build: Design and implement our compiler, runtime, and debugger for PyTorch, TensorFlow, JAX, and MXNet on custom hardware. Optimise Performance: Apply advanced techniques (layout, fusion, scheduling, tiling) to eliminate bottlenecks and maximise throughput. Mentor & Encourage Standards: Lead code reviews, coach peers, and uphold best practices More ❯
Staff Compiler Engineers to architect and build the ML backend (compiler, run-time, and debugger) for our next-generation OPTUs. You will own integration with PyTorch, Tensorflow, JAX, and MXNet down to our low-level kernel drivers. Your mission will be to create seamless support for a broad ecosystem of large AI models, and ensure we are pushing the limits … the OTPU design; ensure software and hardware are designed together to deliver maximum performance. Architect & Build: Design and implement our compiler, runtime, and debugger for PyTorch, TensorFlow, JAX, and MXNet on custom hardware. Optimize Performance: Apply advanced techniques (layout, fusion, scheduling, tiling) to eliminate bottlenecks and maximize throughput. Mentor & Define Standards: Lead code reviews, coach peers, and define best practices More ❯
language - Experience with neural deep learning methods and machine learning - Experience with prompting techniques for LLMs PREFERRED QUALIFICATIONS - Experience with modeling tools such as R, scikit-learn, Spark MLLib, MxNet, Tensorflow, numpy, scipy etc. - Experience with large scale distributed systems such as Hadoop, Spark etc. - PhD in math/statistics/engineering or other equivalent quantitative discipline - Experience with conducting More ❯
and written communication skills Experience working with real-world data sets and building scalable models from big data Experience with modeling tools such as R, scikit-learn, Spark MLLib, MxNet, Tensorflow Experience with large scale distributed systems Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you have a disability and need a workplace accommodation More ❯
in Java, C++, Python, or related language. Experience with neural deep learning methods and machine learning. PREFERRED QUALIFICATIONS Experience with modeling tools such as R, scikit-learn, Spark MLLib, MxNet, Tensorflow, numpy, scipy, etc. Experience with large scale distributed systems such as Hadoop, Spark, etc. Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you More ❯
