d-Matrix

Senior Software Engineer – Kernels

d-Matrix

full-time

Posted on:

Location Type: Hybrid

Location: Bangalore • India

Visit company website

Explore more

✨ AI Apply
Apply

Job Level

Senior

Tech Stack

CloudLinuxPythonPyTorchTensorflow

About the role

  • Be part of the team that helps productize the SW stack for our AI compute engine.
  • Responsible for the development, enhancement, and maintenance of software kernels for next-generation AI hardware.
  • Possess experience building software kernels for HW architectures.
  • Build and scale software deliverables in a tight development window.
  • Collaborate with a team of compiler experts to build out the compiler infrastructure, working closely with other software (ML, systems) and hardware (mixed signal, DSP, CPU) experts in the company.

Requirements

  • MS in computer engineering, math, physics, or a related degree with 5+ years of industry experience or PhD in computer engineering, math, physics, or a related degree with 1+ years of industry experience
  • Strong grasp of computer architecture, data structures, system software, and machine learning fundamentals
  • Proficient in C/C++ and Python development in Linux environments and using standard development tools
  • Experience implementing algorithms in high-level languages such as C/C++ and Python.
  • Experience implementing algorithms for specialized hardware such as FPGAs, DSPs, GPUs, and AI accelerators using libraries such as CUDA, etc.
  • Experience in implementing operators commonly used in ML workloads—GEMMs, Convolutions, BLAS, SIMD operators for operations like softmax, layer normalization, pooling, etc.
  • Experience with development for embedded SIMD vector processors such as Tensilica.
  • Self-motivated team player with a strong sense of ownership and leadership
  • Prior startup, small team, or incubation experience (preferred)
  • Experience with ML frameworks such as TensorFlow and/or PyTorch (preferred)
  • Experience working with ML compilers and algorithms, such as MLIR, LLVM, TVM, Glow, etc. (preferred)
  • Work experience at a cloud provider or AI compute/subsystem company (preferred)
Benefits
  • hybrid, working onsite at our Bangalore, India, offices 3-5 days per week.
Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools
CC++PythonLinuxCUDAFPGAsDSPsGPUsML algorithmsSIMD operators
Soft Skills
self-motivatedteam playerownershipleadership