Description

You will design, implement, and improve machine learning systems and tools to enable research.

Responsibilities

• Develop and maintain deep learning libraries supporting large-scale distributed training.
• Open source high-quality code and create associated documentation.
• Support the community to help users onboard and leverage team solutions.
• Integrate the latest research findings into client products.
• Apply research knowledge and expert coding to platform and framework development.

Required Skills

• 5+ years of professional experience in Python.
• 2+ years of experience with PyTorch.
• 5+ years of experience working with datasets and PyTorch DataLoaders.
• 2+ years of experience developing ML systems in Python or C/C++.
• Experience with ML frameworks including PyTorch or TensorFlow.
• 3+ years of experience with Open Source Software (OSS).
• Solid understanding of algorithms, data structures, and software engineering best practices.
• Familiarity with distributed ML training (FSDP/DDP) is required.

Preferred Skills

• Demonstrated software engineering experience via open-source contributions (e.g., GitHub).

Key Skills

Education

Any Graduate