Senior Software Engineer

Summary

Join OpenTeams, a leading AI company, and become a software engineer specializing in numerical and high-performance computing within our PyTorch team. Contribute to open-source projects like PyTorch, TorchRL, TorchAO, and Triton, utilizing languages such as C++, CUDA, Python, C, and MLIR. Develop new features and performance enhancements for code used by millions. Collaborate with the PyTorch community, users, and clients. Learn from and mentor team members while being part of the Python community. Prior experience with PyTorch or NumPy is a plus but not required; strong communication skills and knowledge of deep learning are valued. OpenTeams offers the chance to grow professionally within the company and the Python open-source community.

Requirements

• Have experience and interest in numerical and high-performance computing
• Have expertise in languages like C++, Python, and CUDA
• Can find your way around large code bases
• Have good communication skills

Responsibilities

• Work with modern numerical computing and deep learning technology
• Contribute to open-source projects like PyTorch, TorchRL, TorchAO, and Triton
• Use C++, CUDA, Python, C, and MLIR
• Develop new features, performance enhancements and help maintain code with millions of users
• Interact with the PyTorch development community, users, and our clients who are building PyTorch and building with PyTorch
• Collaborate with, learn from, and mentor team members
• Be a part of the Python community

Preferred Qualifications

• Prior experience with particular packages like PyTorch or NumPy
• Knowledge of deep learning