Description

We are looking for an engineer for Machine Learning teams at Tzafon, a company developing advanced computational systems for training foundation models at massive scale. You will be responsible for designing and running large training runs, building distributed training infrastructure, developing post-training pipelines, and writing custom kernels in CUDA and Triton. Required: deep experience training foundation models on large clusters, deep proficiency in Python and PyTorch, expertise in tensor parallelism and FSDP, and the ability to manage stable training jobs for weeks while debugging distributed issues. Location: Tel Aviv · Remote work · Full-time · Department: Machine Learning team To apply: On the company website, via the button below

Details

Location

Tel Aviv