Neural Network Optimization Engineer

HRB; Canada; Remote; CAD 80,000 - 120,000; Full time

Job summary

An innovative company is seeking a Neural Network Optimization Engineer to join a talented team dedicated to pushing the boundaries of AI technology. In this role, you will design and optimize neural networks that leverage cutting-edge AI accelerator chips, ensuring that our products meet the highest performance standards. You will work closely with hardware engineers and software teams to address real-world challenges and enhance customer experiences. If you are passionate about AI and thrive in collaborative environments, this opportunity is perfect for you to make a significant impact in the tech industry.

Qualifications

• 5+ years of software development experience, specifically in AI.
• Expertise in developing neural networks for high-performance AI chips.

Responsibilities

• Develop and deliver customer neural networks and applications.
• Evaluate neural network performance through profiling and benchmarking.
• Collaborate with cross-functional teams to resolve challenges.

Skills

AI Enthusiasm

Neural Network Optimization

C++ Development

Software Development

Problem-Solving

Teamwork

Education

Degree in Computer Science

Degree in Engineering

Tools

Profiling Tools

Benchmarking Tools

Job description

We’re looking for best-in-class engineers to join our existing top-notch team. When you join, you will be part of a team that designs, develops, and verifies the software that interacts with our chip, collaborating with hardware engineers and fellow software engineers. By creating software that fully realizes the capabilities of the hardware, you will help bring AI inference to the general populace.

As part of this exceptional team, you are able to — and get excited by — identifying functional and performance bottlenecks and alleviating them to achieve scalable and reliable software. You excel in environments with complex software and hardware designs.

The Neural Network Optimization Engineer is responsible for designing, developing, and optimizing cutting-edge neural networks for our customers. The role involves hands-on neural network development, working with a cross-functional team focused on customer needs, and acting as a conduit between customers and our software and hardware teams. Using and/or developing our software tools (such as kernel support and physical allocation), especially for customer neural nets requiring enhanced functionality or performance, ensures optimal utilization of our technology.

The engineer understands our products, technology, and applications, and is an expert in developing neural networks for high-performance AI accelerator chips. They diagnose and address real-world challenges, resolve issues, escalate as needed, and provide feedback to improve existing products and inform next-generation development.

Responsibilities

• Use and advance our products in developing and delivering leading-edge customer neural networks and applications.
• Evaluate neural network performance at chip, system, and application levels through profiling and benchmarking.
• Identify and diagnose challenges in using our products for specific use cases and develop projects to resolve them.
• Engage with the Customer Networks Lead on escalations to the software and hardware teams.
• Support customers post-sale to ensure successful product delivery.
• Collaborate with the Products team to incorporate customer feedback for seamless interactions.
• Manage program/project teams to deliver Neural Network Model Garden models, including designing and implementing low-level C++ kernels for various neural network operations.
• Develop complex kernels and compiler strategies as needed to meet network requirements.
• Communicate performance optimization ideas to compiler/kernel engineers and future product architects.

Preferred Skills & Experience

• Enthusiastic and knowledgeable about all things AI.
• Degree in Computer Science, Engineering, or related field.
• At least 5 years of software development experience, specifically in AI.
• Knowledge of neural network operators such as convolutions, transformers, RNNs.
• Experience with end-to-end software development, including data structures, algorithms, and software architecture patterns.
• Experience working in cross-functional teams.
• Strong teamwork and problem-solving skills.
• Experience tuning and optimizing neural networks and code for high performance.
• Excellent communication skills and experience in program/project management.
• Superior technical and interpersonal problem-solving skills.