- Experience in ML frameworks such as PyTorch,JAX or Tensorflow.
- Experience with distributed training and distributed training framework such as DeepSpeed.
- Experience with LLM or Vision,especially large model is a plus.
- Experience with ML pipeline. Strong communication and problem-solving skills