Remote Software Engineer

Platform Recruitment

London, UK

about 2 months ago

Role details

Contract type

Permanent contract

Employment type

Full-time (> 32 hours)

Experience level

Experienced

Experience required

3 years minimum

Compensation

£100,000.0

Working hours

Regular working hours

Job source

find.jobs

Tech stack

Artificial Intelligence Nvidia CUDA OpenCL Tensorflow Pytorch Deep Learning ONNX (Open Neural Network Exchange) Format Hardware Acceleration

Job description

This position is best described as an AI Kernel / GPU Kernel Engineer position. You will make a significant impact on the business by owning the design and implementation of novel AI Kernels for deep learning operations. You will optimise hardware accelerators in collaboration with deep learning researchers, translating their high-level models into low-level, high-performance code.

Requirements

3+ years’ experience in GPU development
Experience developing GPU/AI Kernels
Experience engineering for AI/ML workloads
Experience with any of: CUDA, OpenCL, HIP, SYCL
Knowledge of deep learning algorithms
Interested in optimising tough linear algebra equations
Knowledge of AI framework internals (PyTorch, TensorFlow, ONNX etc)

About the company

The AI R&D arm of a major chip subsidiary are looking for a Software Engineer to join the AI Kernel team. Comprised of a team of elite global technical experts in Compiler, AI and GPU research, you will work on some of most cutting edge developments in the AI acceleration world - some of which are so advanced they still years from being made public. Contributing significantly to the open-source community, you will make an impact on the wider tech world.