ML Systems/Infrastructure Engineer

London Office
Hybrid
ML Systems/Infrastructure Engineer
Oriole is seeking a talented ML Systems/Infrastructure Engineer to help co-optimize our AI/ML software stack with cutting-edge network hardware. You'll be a key contributor to a high-impact, agile team focused on integrating middleware communication libraries and modelling the performance of large-scale AI/ML workloads.
Key Responsibilities
Design and optimize custom GPU communication kernels to enhance performance and scalability across multi-node environments
Develop and maintain distributed communication frameworks for large-scale deep learning models, ensuring efficient parallelization and optimal resource utilization.
Profile, benchmark, and debug GPU applications to identify and resolve bottlenecks in communication and computation pipelines.
Collaborate closely with hardware and software teams to integrate optimized kernels with Oriole's next-generation network hardware and software stack.
Contribu...