AI Infrastructure Software Engineer

JD.com International Limited; Greater London; On-site; GBP 90,000 - 130,000; Full time

Job summary

JD.com International Limited is expanding its international footprint with a focus on AI infrastructure across cross-border e-commerce. You will help design and implement scalable architectures for GPU-accelerated workloads, leveraging PyTorch, Triton, and advanced scheduling tools like Kubernetes or Ray.

Ideal candidates bring 3+ years in AI infrastructure, strong Python skills, and hands-on CUDA/C++ experience, with a degree in CS/EE/Applied Math.

Qualifications

• 3+ years in AI infrastructure, distributed systems, HPC, or large-scale cloud platform development.

Responsibilities

• Participate in building AI infrastructure architecture for cross-border e-commerce projects.

Skills

AI infrastructure

Python

CUDA/C++

GPU architecture

Megatron-LM/DeepSpeed

Kubernetes / Ray

Education

Bachelor’s degree in CS/EE/Applied Math

Tools

CUDA/C++

Triton

PyTorch

Kubernetes

Ray

Job description

Who We Are

At JD.com, also known as JINGDONG, we are building one of the worlds most trusted technology and supply chain ecosystems. What started as a bold idea in China has evolved into a NASDAQ-listed, Fortune 44 global enterprise serving over 700 million active customers. Powered by advanced logistics, innovative technology, and a commitment to integrity, JD.com operates across retail, logistics, technology, property and more, creating solutions that make everyday life smarter, faster, and more connected.

Our Global Reach

JD.com is expanding its international footprint through a digitally intelligent, cross-border supply chain and world-class logistics network, supported by platforms such as Joybuy, which brings trusted products to customers worldwide.

Today, our teams operate across China, the U.S., U.K., Netherlands, France, Germany, Spain, Brazil, Hungary, Japan, South Korea, Australia, Thailand, Vietnam, Malaysia, Indonesia, Saudi Arabia, the UAE and beyond, reaching customers in every corner of the world.

Learn more about who we are and what we do: https://corporate.jd.com/

What We Offer

Joining JD.com means becoming part of a company where people, innovation, and impact go hand in hand. We offer a competitive salary, a great location, and a supportive, people-first culture where your growth truly matters. With access to global projects, learning opportunities, and an inclusive environment that values every voice, you'll have the space to shape the future with us.

About The Team

You'll be joining the JD.com team, which values collaboration, ownership, and a commitment to delivering high-quality results in a fast-paced, dynamic environment.

Responsibilities & Objectives

• Participate in building the AI Infrastructure technical architecture for JD.com's cross-border e-commerce; develop and drive the implementation of architectural standards.
• Lead the top-level architectural design for complex technical projects; evaluate critical technical issues and break down project execution steps.
• Scheduling System Optimisation: Participate in the R&D of the intelligent computing operating system; Optimize GPU resource scheduling capabilities within Kubernetes or Ray.
• High Availability Assurance: Resolve GPU cluster fault recovery (Fault Tolerance) and elastic fault tolerance (elastic checkpointing).
• Efficiency & Observability: Build cluster efficiency evaluation models; precisely measure computing resource utilisation (MFU/HFU).
• Track industry trends and cutting-edge technologies; introduce new technologies and solutions to the team based on actual business needs.

Requirements

• 3+ years of experience in AI infrastructure, distributed systems, High-Performance Computing (HPC), or large-scale cloud platform development.
• Proficient in Python, with a solid foundation in data structures and algorithms, and a rigorous coding style.
• Proficient in GPU/NPU architecture, with hands-on development and optimisation experience in CUDA C/C++ or Triton.
• Strong self-drive and technical curiosity; ability to proactively track the most cutting-edge AI Infra trends in the industry.
• Bachelor's degree or higher in Computer Science, Electronic Information, Automation, Applied Mathematics, or related fields; experience in cross-border e-commerce is highly preferred.
• Deep understanding of the underlying source code and runtime mechanisms of PyTorch, Megatron-LM, or DeepSpeed.

At JD.com, we're committed to building a diverse and inclusive workplace where everyone can thrive. We're proud to be an equal opportunity employer and make all employment decisions without regard to race, color, religion, gender, gender identity or expression, sexual orientation, national origin, disability, age, veteran status, or any other protected characteristic.

About us

What started as one bold idea in China has grown into one of the worlds most trusted technology and supply chain businesses. Today, JD.com (JINGDONG) serves more than700 million customers, using technology, logistics and innovation to create smarter, faster and more connected experiences. From retail and logistics to technology, healthcare and beyond, we're solving real-world challenges and making everyday life easier.

As we continue to grow, so do the opportunities for the people who grow with us.

Our global business Our teams work across 20+ countries and regions, building smarter supply chains and helping businesses and communities connect across borders. From technology hubs and retail operations to logistics networks and corporate teams, we're creating opportunities to learn, collaborate and make an impact on a global scale. Whether you're joining us in Europe, the Middle East, Asia or the Americas, you'll be part of one global team working towards a shared purpose.

So let's do this. Let's deliver joy.