AI21 Labs is hiring a Deep Learning Engineer to build and maintain its LLM training infrastructure, optimize training and serving performance, adapt models to new GPU hardware, and run training/evaluation experiments. Core stack: Python, PyTorch, distributed training frameworks (FSDP/Megatron/DeepSpeed), and inference systems like vLLM.
Sign in to see your match