Description

You will engineer, design, and improve highly scalable machine learning systems and tools to enable research.

Responsibilities

• Apply research knowledge and expert coding skills to platform and framework development projects.
• Write clean and robust machine learning code.
• Create documentation and support users onboarding to team libraries and tools.

Required Skills

• 5+ years of Python experience.
• 2+ years of PyTorch experience.
• 5+ years of Dataset / PyTorch DataLoader experience.
• 3+ years of OSS experience.
• Demonstrated software engineering experience via work or open-source contributions.
• 2+ years of experience developing machine learning algorithms in Python or C/C++.
• Experience with machine learning frameworks such as PyTorch or TensorFlow.
• Experience working with large datasets and data pipelines.

Preferred Skills

• 0-2 years of Distributed ML Training (FSDP/DDP) experience.

Key Skills

Education

Bachelor's degree