Deep Learning Team Lead

Job description preview

From AI21 Labs’s original posting

Description AI21 is looking for a Deep Learning Engineer – Team Lead to join our team. AI21 is one of the few companies in the world to have trained multi-billion-parameter Large Language Models (LLMs). Building and serving models at this scale requires strong software engineering, deep learning expertise, and cutting-edge infrastructure. In this role, you will lead a team of engineers working on our LLM training and inference infrastructure. You will collaborate closely with applied researchers to develop and scale new training methods, improve performance, optimize model serving, and make the best use of advanced hardware accelerators. This is a hands-on technical leadership role combining engineering, research, and team management. Role and Responsibilities • Lead, mentor, and manage a team of deep learning engineers, including task delegation, technical guidance, and project execution. • Develop, improve, and maintain our LLM training framework and supporting infrastructure. • Optimize training performance, scalability, and model quality through parallelism techniques, training SDK improvements, and new model architectures. • Research, develop, test, and integrate new training…