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

TikTok
US - California - San Jose
View Company Profile / << Go Back

  • Job Type: Full time
  • 3 days ago

Job Description

Build and optimize infrastructure for large-scale model training and online inference for recommendation systems and LLMs. Develop distributed systems and implement GPU optimizations to improve system performance and scalability.

Requirements: Requires a Bachelor's degree in Computer Science or a related field with strong programming skills in C++ or Python. Candidates should have knowledge of Transformer architectures, LLMs, and familiarity with PyTorch or TensorFlow.

Key Skills: C++, Python, PyTorch, TensorFlow, Distributed Computing, GPU Optimization, CUDA, Triton, Transformer Architectures, LLM Infrastructure, Data Structures, Algorithms

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




Fast Track Upload