What is the role of a Senior Deep Learning Compiler Engineer - XLA at Jobgether?

The Senior Deep Learning Compiler Engineer - XLA position at Jobgether is a Full-time or part-time position opportunity in the Information Technology field.

Where is this Senior Deep Learning Compiler Engineer - XLA job located?

United States, Other / Non-US, United States

What type of employment is offered for this Senior Deep Learning Compiler Engineer - XLA role?

Full-time or part-time position

What is the expected salary for this Senior Deep Learning Compiler Engineer - XLA job?

Compensation will be discussed during the hiring process.

How can I apply for the Senior Deep Learning Compiler Engineer - XLA position at Jobgether?

You can apply directly through the application link provided.

Senior Deep Learning Compiler Engineer - XLA at Jobgether | Jobs and Employment

Senior Deep Learning Compiler Engineer - XLA

This position is posted by Jobgether on behalf of a partner company. We are currently looking for a Senior Deep Learning Compiler Engineer - XLA in United States.

This role focuses on developing and optimizing compilers for high-performance deep learning workloads on modern GPU architectures. You will design, implement, and tune advanced compiler optimization algorithms to accelerate training and inference for deep learning frameworks at scale. The position involves collaborating with framework teams, hardware engineers, and cross-functional partners to deliver production-grade software that powers next-generation AI systems. You will work on graph partitioning, tensor sharding, performance analysis, and code generation, while also contributing to user-facing library features. This role combines deep technical expertise, creativity, and autonomy in a dynamic, research-driven environment with significant impact on AI computing performance.

Accountabilities:

Develop compiler optimization techniques for deep learning network graphs
Design and implement graph partitioning and tensor sharding for distributed training and inference
Optimize performance and analyze computational efficiency on GPU hardware
Generate code for NVIDIA GPU backends using MLIR, LLVM, OpenAI Triton, or similar compilers
Contribute to user-facing features in deep learning frameworks and related libraries
Collaborate with GPU hardware teams to align software features with next-generation architectures
Mentor junior engineers and support knowledge sharing within the team

Requirements:

Bachelors, Masters, or Ph.D. in Computer Science, Computer Engineering, or related field, or equivalent experience
4+ years of experience in performance analysis, compiler optimization, or deep learning software development
Strong C/C++ programming skills with expertise in software design, debugging, and testing
Knowledge of CPU, GPU, or other high-performance hardware architectures, including distributed computing
Experience with CUDA or OpenCL is desirable
Familiarity with XLA, TVM, MLIR, LLVM, OpenAI Triton, or deep learning frameworks such as JAX, PyTorch, or TensorFlow is a strong plus
Ability to work independently, define project scope, and deliver high-quality software
Excellent communication and collaboration skills; experience mentoring junior engineers is a bonus

Benefits:

Competitive salary range ($152,000$241,500 USD)
Eligibility for equity, bonuses, and comprehensive benefits packages
Remote work eligibility with flexibility depending on location
Opportunities for professional growth, mentorship, and career development
Dynamic, research-driven environment working on cutting-edge AI technology
Inclusive, diverse, and collaborative workplace culture

Why Apply Through Jobgether?

We use an AI-powered matching process to ensure your application is reviewed quickly, objectively, and fairly against the role's core requirements. Our system identifies the top-fitting candidates, and this shortlist is then shared directly with the hiring company. The final decision and next steps (interviews, assessments) are managed by their internal team.

We appreciate your interest and wish you the best!

Why Apply Through Jobgether?

Data Privacy Notice: By submitting your application, you acknowledge that Jobgether will process your personal data to evaluate your candidacy and share relevant information with the hiring employer. This processing is based on legitimate interest and pre-contractual measures under applicable data protection laws (including GDPR). You may exercise your rights (access, rectification, erasure, objection) at any time.

#LI-CL1

Senior Deep Learning Compiler Engineer - XLA at Jobgether – United States

About This Position

Scan to Apply

Job Location

Frequently asked questions about this position