【AI Engineer】HPC & Software Acceleration Specialist at a TSE Prime-listed Industry Leader | Hybrid

★ AI Compiler Engineer | Multi-core processor-related company

• Business Level Japanese Required

◆ Hybrid Work

◆ Listed Company

◆ Global Business

◆ Annual Salary： 6 million ～ 10 million yen.

-------------【About the company】-------------

The company is a software company whose corporate message is "Speed up your Business.

They accelerate their customers' businesses in various fields such as medical, manufacturing, finance, and entertainment by parallelizing and optimizing software to efficiently utilize multi-core processors and accelerating applications using new memory technology that realizes power saving and high-speed IO, thereby achieving green IT.

-------------【 Job Description】-------------

【Overview】

Machine learning is currently used in a variety of applications.

Various AI accelerators are being developed to speed up or reduce the power consumption of these applications.

The company is developing an AI compiler framework for customers developing AI accelerators.

The AI Compiler automatically applies transformations and code generation to accelerate the training and inference workloads of machine learning models on the target AI accelerator.

The company is looking for an engineer to design and develop this AI compiler framework.

In this role, you will design methods for quantization, computational graph optimization, loop and memory placement optimization for the target AI accelerator, as well as implementation of these methods in the compiler.

You will also research the latest research and OSS trends for AI accelerators and AI compilers, and will be involved in technology selection using those research and OSS trends. In addition, you will also collaborate with hardware designers and contribute to the design of next-generation AI accelerators by designing hardware and software in a collaborative manner.

Why not join the project to work on the development of a large-scale, state-of-the-art AI compiler framework and next-generation AI accelerator?

【Specifically】

・Design and development of AI compiler framework

・Design of acceleration techniques for target AI accelerators

・R&D and implementation of AI model compression techniques such as quantization and branch pruning

・Acceleration of machine learning models using the AI compiler framework

・Propose improvement methods for the hardware architecture of AI accelerators

【Scope of change of business to be engaged in】

General operations as stipulated by the company

【Project Satisfaction】

・Challenge new technological innovations in the rapidly growing AI/ML market.

・Engage in the development of cutting-edge, large-scale computer systems related to AI compilers and AI frameworks.

・Develop and contribute to the development of well-known OSS

・Design and develop the next generation of AI accelerators

【Development Environment】

・Development Environment: C/C++, Python

・Compiler Frameworks: MLIR, LLVM, TVM

・AI frameworks: PyTorch, TensorFlow, ONNX

・Other development environment: Linux

・Development Support Tools: Git, GitLab, GitHub

・Development Methodology: Ticket Driven Development, Scrum Development

・Development Content Type: B2B

-------------【 Requirements】-------------

■Required

・Deep knowledge and experience with compiler design and optimization techniques

・Programming knowledge and experience using C/C++ and Python

・Deep knowledge of computer science

・Interest in the field

■Preferred

・Familiarity with OSS compilers such as MLIR, LLVM, TVM, etc.

・Familiarity with deep learning frameworks such as PyTorch, TensorFlow, ONNX, etc.

・Knowledge of hardware architectures such as CPUs, GPUs, FPGAs, AI accelerators, etc.

・Experience with high-performance programming

・Knowledge and R&D experience in model compression techniques such as quantization and pruning

・Experience in developing and contributing to OSS (Open Sourcre Software)

■Ideal Applicants

・Have an interest in the field and want to contribute to solving problems in the field.

・Able to identify and define issues based on the current state of society and projects

・Able to devise and propose solutions to the issues found.

・Contribute to teamwork through leadership and communication skills

・Able to use advanced programming skills to create products and solutions that solve problems.

・Keep up to date with the latest OSS implementations, papers, books, etc. in the field and contribute them to the project

--------------------------------------------------

【Working Hours】

Discretionary Labor System

【Holidays】

・Saturday/Sunday/National Holiday

・Annual Paid Leave

・New Year Holiday

【Welfare】

・Full Social Insurance

・Commuting Allowance

・Child Allowance

・Recreation Assistance

・Training/Self Development Allowance

・Outside resident allowance of 30,000 yen/month

・Travel and accommodation expenses for company trips

・Company expenses for social gatherings

・Maintenance fee for club activities (e.g., ski club: 60,000 yen per person per year; other clubs include golf, futsal, and board game clubs)

-------------------------------------------------------------------------