Deep Learning Engineer

AI21 is seeking a Deep Learning Engineer to improve and maintain its training infrastructure for multi-billion parameter Large Language Models. The role involves developing and scaling new training methods, optimizing model serving for high throughput and low latency, and running experiments to optimize LLMs across various tasks. This position requires strong engineering fundamentals and a deep understanding of modern LLM training architectures and distributed systems.