AI Infra Engineer Graduate (Recommendation & LLM) 2027 Start (PhD)

Job Description

Build and optimize infrastructure for large-scale model training and online inference for recommendation systems and LLMs. Collaborate with researchers to implement GPU optimizations and resolve system bottlenecks to improve performance.

Requirements: Requires a PhD in Computer Science or a related field with strong programming skills in C++ or Python. Candidates should have a deep understanding of Transformer architectures, LLMs, and distributed training concepts.

Key Skills: C++, Python, PyTorch, TensorFlow, Distributed Systems, GPU Optimization, CUDA, Triton, LLM Infrastructure, Transformer Architectures, Data Structures, Algorithms, KV Cache, Continuous Batching, FlashAttention, FSDP

Benefits: Medical Insurance, Dental Insurance, Vision Insurance, 401(k) Savings Plan With Company Match, Paid Parental Leave, Short-term Disability Coverage, Long-term Disability Coverage, Life Insurance, Wellbeing Benefits, Paid Holidays, Paid Sick Days, Paid Personal Time