AI/ML Application Performance Engineer

ZipRecruiter; Chesterbrook (Chester County); Remote; USD 90,000 - 150,000; Full time

Job summary

An innovative firm is seeking a talented AI/ML Application Performance Engineer to drive advancements in high-performance networking solutions. In this exciting role, you will leverage your expertise in AI and Machine Learning to optimize applications and enhance performance across diverse use cases. You will work alongside industry experts and contribute to groundbreaking technologies in the semiconductor industry. With a flexible remote work environment and opportunities for collaboration, this position offers a chance to make a significant impact while enjoying a competitive compensation package and comprehensive benefits.

Benefits

Medical, dental, and vision coverage

401(k) with company match

Open Time Off (OTO)

Paid holidays

Life insurance

Dependent care flexible spending account

Accidental injury insurance

Pet insurance

Qualifications

• 3-5 years experience with HPC and AI/ML applications on clusters.
• Extensive experience with Python and shell scripting.

Responsibilities

• Benchmark and optimize AI/ML applications on HPC hardware.
• Collaborate with teams to improve performance and capabilities.

Skills

AI/ML Application Benchmarking

Performance Optimization

Network Analysis

Communication Skills

Python Programming

Shell Scripting

HPC Resource Management

Message Passing Interface (MPI)

Education

Bachelor's degree in Computer Science

Master's degree in related field

Tools

NVIDIA Nsight Systems

NeMo

PyTorch Lightning

Megatron-LM

DeepSpeed

SLURM

PBS

Job description

Job Description

Salary:

Cornelis Networks delivers the world's highest performance scale-out networking solutions for AI and HPC datacenters. Our differentiated architecture seamlessly integrates hardware, software, and system-level technologies to maximize the efficiency of GPU, CPU, and accelerator-based compute clusters at any scale. Our solutions drive breakthroughs in AI & HPC workloads, empowering our customers to push the boundaries of innovation. Backed by top-tier venture capital and strategic investors, we are committed to innovation, performance, and scalability - solving the world's most demanding computational challenges with our next-generation networking solutions.

We are a fast-growing, forward-thinking team of architects, engineers, and business professionals with a proven track record of building successful products and companies. As a global organization, our team spans multiple U.S. states and six countries, and we continue to expand with exceptional talent in onsite, hybrid, and fully remote roles.

Cornelis Networks is hiring a talented AI/ML Application Performance Engineer to help drive innovation and contribute to the development of cutting-edge technologies in the semiconductor industry. In this role, you will be responsible for providing technical expertise in AI and Machine Learning (ML) that can be applied to a diverse range of AI/ML use cases, working alongside a team of industry experts to shape the future of high-performance networking solutions.

Key Responsibilities:

1. Perform benchmarking and optimization of open source and industry-standard AI/ML applications with current and future HPC hardware.
2. Develop, execute, and maintain software required to run AI/ML applications and benchmarks.
3. Participate in the development of supporting libraries and middleware.
4. Assist sales and marketing teams by delivering proof points and performance benchmarking comparisons between Cornelis Omni-Path and competing interconnects.
5. Collect and analyze performance data, identifying performance limitations, and determining the best approach and techniques to improve performance.
6. Present research findings both within the company and to external stakeholders.
7. Collaborate with cross-functional teams across all levels of a corporation to evangelize the capabilities and performance advantages of Cornelis products.

Qualifications:

1. Knowledge of HPC resource management and job scheduling systems (e.g., SLURM, PBS).
2. Hands-on experience with analyzing and optimizing networks to improve scale-out performance using a range of profiling tools such as NVIDIA Nsight Systems.
3. Experience with AI frameworks like NeMo, PyTorch Lightning, Megatron-LM, and DeepSpeed.

Minimum Qualifications:

1. Bachelor's degree (Master's preferred) in computer science, engineering, math, or related technical discipline.
2. 3-5 years of experience running HPC and/or AI/ML applications on clusters.
3. Ability to set up, run, and analyze AI/ML application benchmarks and demonstrate a proficient understanding in message passing, scaling optimization, and identifying performance bottlenecks.
4. Ability to modify AI/ML models and distribute training across networks, outside of a single GPU compute platform.
5. Experience with Message Passing Interface (MPI) and compiling software with a variety of compilers (Intel, gcc, etc.) and libraries.
6. Extensive Python and shell script experience.
7. Experience with HPC network architectures such as Omni-Path, InfiniBand, or Ethernet.
8. Experience operating in UNIX or Linux computing environment.
9. Excellent written and verbal communication skills.

Location: This role fully supports remote work for employees residing within the United States, with the flexibility to travel to our Chesterbrook Corporate Center located in Wayne, PA occasionally for in-person collaboration.

We offer a competitive compensation package that includes equity, cash, and incentives, along with health and retirement benefits. Our dynamic, flexible work environment provides the opportunity to collaborate with some of the most influential names in the semiconductor industry.

At Cornelis Networks, your base salary is only one component of your comprehensive total rewards package. Your base pay will be determined by factors such as your skills, qualifications, experience, and location relative to the hiring range for the position. Depending on your role, you may also be eligible for performance-based incentives, including an annual bonus or sales incentives.

In addition to your base pay, you'll have access to a broad range of benefits, including medical, dental, and vision coverage, as well as life insurance, a dependent care flexible spending account, accidental injury insurance, and pet insurance. We also offer generous paid holidays, 401(k) with company match, and Open Time Off (OTO) for regular full-time exempt employees. Other paid time off benefits include sick time, bonding leave, and leave.

Cornelis Networks does not accept unsolicited resumes from headhunters, recruitment agencies, or recruitment services. Cornelis Networks is an equal opportunity employer, and all qualified applicants will receive consideration for employment without regard to race, color, religion, gender, gender identity or expression, sexual orientation, national origin, disability, veteran status, genetic information, or any other characteristic protected by law. We encourage applications from all qualified candidates and will accommodate applicants' needs under the respective laws throughout all stages of the recruitment and selection process.