Senior HPC and AI Network Software Architect

This is a fully remote position, open to applicants in Poland.

📋 Description

• Design and enhance the architecture of scalable software systems for distributed AI training and inference.

• Create and assess next-generation communication and runtime functionalities in libraries such as NCCL, UCX, and UCC.

• Collaborate with AI framework teams (such as TensorFlow, PyTorch, JAX) to establish integrations and investigate new methodologies.

• Work together on hardware and system-level features involving GPUs, DPUs, and interconnects.

• Foster innovation in runtime systems, communication libraries, and AI-specific protocol layers.

⛳️ Requirements

• Ph.D. or equivalent industry experience in computer science, computer engineering, or a closely related discipline.

• Over 5 years of experience in systems programming, parallel or distributed computing, high-performance networking, or large-scale data transfer.

• Strong programming expertise in C++, Python, and ideally CUDA or other GPU programming paradigms.

• Extensive practical experience with AI frameworks (e.g., PyTorch, TensorFlow, JAX).

• Proven track record in developing and optimizing high-throughput, low-latency systems.

• Excellent collaboration abilities in a multinational, interdisciplinary environment.

🏝️ Benefits

• Comprehensive benefits package

• Highly competitive salaries