Sr. Staff Software Engineer, AI Infra

LinkedIn, the world's largest professional network, is seeking a Sr. Staff Software Engineer to join their AI Infrastructure team. This role presents an exciting opportunity to push the boundaries of scaling large models and shape the future of AI platforms. The position focuses on scaling LinkedIn's AI model training, feature engineering, and serving infrastructure handling models with hundreds of billions of parameters.

The team is deeply involved in optimizing performance across algorithms, AI frameworks, data infrastructure, compute software, and hardware to maximize the potential of their extensive GPU fleet. They work closely with the open source community and have many committers to projects like TensorFlow, Horovod, Ray, vLLM, and Huggingface.

As a Sr. Staff Software Engineer, you'll have the opportunity to work on cutting-edge technologies including LLMs, GNNs, Incremental Learning, and Online Learning. You'll be responsible for building next-gen training infrastructure, implementing high-performance data I/O, and enabling distributed training for massive models. The role involves sophisticated technical leadership, requiring both deep technical expertise and the ability to mentor others.

The position offers a hybrid work arrangement, combining remote work with office presence, and competitive compensation. You'll be part of a team that values innovation, collaboration, and technical excellence, working on problems at the forefront of AI infrastructure development. This is an ideal role for someone who wants to make a significant impact on the scale and efficiency of AI systems while working with industry-leading technology.