Job Description

Your Responsibilities:

• Constructing scalable and resilient training and inference pipelines for deep learning.
• Delving into the inner workings of open-source deep learning frameworks to enhance their capabilities.
• Identifying and resolving performance bottlenecks.
• Collaborating closely with researchers and fellow engineers.
• Developing a comprehensive understanding of trading systems.

Requirements:

• Proficiency in the inner workings of deep-learning frameworks such as PyTorch, JAX, TensorFlow, etc.
• Thorough knowledge of computer architecture.
• Proficiency in programming with C++ and Python.

Preferred Qualifications:

• Experience with the JAX ecosystem, including XLA, Flax, etc.
• Proficiency in programming for GPUs or other accelerators like CUDA, Triton, Pallas, etc.
• Experience in Linux system programming.
• Familiarity with large-scale distributed training.
• Contributions to open-source projects in the realm of data science and machine learning.

Job Tags

Similar Jobs