AI ML Software Engineer

X-Scalesolutions; Columbus (OH); Hybrid; USD 110,000 - 160,000; Full time

Benefits offered by this job

Healthcare

Retirement plan

PTO

Job summary

X-ScaleSolutions in Columbus, Ohio, is seeking an enthusiastic AI ML/DL Software Engineer to design and develop a production‑quality AI software stack for diverse HPC systems, using PyTorch, DeepSpeed, and related frameworks. The role focuses on training and inference across vision, language, and other models, with a lead project responsibility.

The ideal candidate has strong Linux and Python skills, experience with AI models and HPC platforms, and excellent collaboration abilities.

Qualifications

• Strong experience in Linux, including troubleshooting and performance tuning.
• Proficient Python programming for AI/ML workflows and tooling.
• Hands-on experience with AI models and frameworks (design and debugging).
• Knowledge of HPC technologies and platforms and related workflows.
• Excellent communication and collaboration with cross‑functional teams.

Responsibilities

• Lead design and development of production‑quality AI DL/ML software on HPC systems.
• Collaborate across hardware, software, and data teams to deliver scalable models.
• Contribute to performance tuning, validation, and deployment of AI solutions.

Skills

Linux

Python

AI Models

HPC

Communication

Job description

X-ScaleSolutions is a start-up based in Columbus, Ohio. We specialize in a range of high-performance and scalable solutions for current generation systems ranging from small deployments of a couple systems to multi-petaflop systems and the emerging Exascale systems. The mission of the company is to develop innovative and leading-edge software products, with a focus on five areas: 1) High-Performance Computing, 2) AI (Deep/Machine Learning), 3) Big Data, 4) Cloud Computing, and 5) HPC Security.

Are you interested in working on state-of-the-art hardware, software and middleware?

X-ScaleSolutions is looking for an enthusiastic AI ML/DL Software Engineer with desired experience in various AI Models (Vision, Language, and others), Deep Learning Frameworks (PyTorch, DeepSpeed), Machine Learning Frameworks, High-Performance and Scalable execution (Training and Inference) of the AI models with various DL/ML frameworks on modern HPC systems with CPUs and GPUs, and Hyperparameter optimization to join our team. The AI ML software engineer is expected to take a lead role in projects to design and develop production quality AI DL/ML software stack with novel features and capabilities for a range of applications.

Requirements:

• Strong experience in Linux
• Strong experience in Python programming
• Strong experience in AI Models and Frameworks (design and debugging)
• Experience on HPC technologies and platforms
• Excellent communication and collaboration skills

The ideal candidate will have:

• Multiple years of experience in designing, developing and execution of AI models
  and frameworks
• Good knowledge of HPC, multi-core processors, GPUs/accelerators, and high-performance interconnects (InfiniBand, High Speed Ethernet, Slingshot, Omni-Path, or similar high-performance networking technologies).
• Familiar with the typical design cycle (design, develop, test, tune, release, bug fixes, etc.) for producing and maintaining high quality AI software stack and the associated tools
• Demonstrate strong analytical, troubleshooting and problem-solving skills
• Experience with training real-world AI models in at least one modality (vision, language, audio, etc)

This is a full-time position, though part-time candidates will be considered. Flexible
schedule and remote position. Healthcare, retirement and PTO package provided. Must
be currently eligible to work in the US.

Job Type: Full-time or Part-time

Experience:

• Training and Inference with various AI Models, Frameworks, and Applications: 2
  years (Required)
• High Performance Computing Technologies and Platforms: 1 year (Required)
• Will consider entry level through senior level applicants.

Work Location: Remote