Job Description:
We are looking for highly motivated individuals to join our team as Machine Learning Platform Interns. The successful candidate will be responsible for working on a machine learning platform. The role is an engineering role and requires a good understanding of machine learning concepts and programming skills.
Responsibilities:
- Large scale machine training and inference workload performance tunning in AI platform
- Write clean, efficient, and well-documented code
- Collaborate with team members to ensure successful delivery of projects
- Participate in team meetings and contribute to team activities
Requirements:
- Currently pursuing a Bachelor's or Master's degree or PhD in Computer Science or a related field
- Strong programming skills in Python, Linux shell
- Good understanding of machine learning concepts and algorithms
- Familiarity with one or more machine learning libraries such as Scikit-Learn, TensorFlow, or Keras. Knowledge/Experience with Deepspeed/Megatron is a plus.
- Ability to work independently and as part of a team
- Excellent problem-solving and analytical skills
- Strong communication
- Experience or knowledge with LLM is a plus