Software Engineer- Machine Learning + TVM

Job Title: Software Engineer

Location: Hyderabad

Job Type: Full-Time

Educational Qualification: B.E, B.Tech, M.E, MTech Only be considered (Mechanical and Civil graduates are not considered)

Mandatory Skills

• Strong hands-on experience in C/C++, Python, NumPy
• Experience with Machine Learning (ML) frameworks: TensorFlow, PyTorch, ONNX
• Proficiency in TVM (Tensor Virtual Machine)
• Knowledge of OpenCL C++

Required Knowledge, Skills, and Abilities

• Proficiency in C, C++, Python, and ML libraries
• Solid foundation in Linear Algebra
• Understanding of Neural Network optimization, graph lowering, and fine-tuning
• Strong analytical and debugging skills
• Sound knowledge of Object-Oriented Programming (OOP) and Software Design Patterns
• Expertise in TVM Framework
• Basic understanding or experience with hardware architecture is a plus
• Experience with full-stack framework development, such as:
• Multimedia frameworks (e.g., GStreamer, OpenVx, OpenMax)
• Graphics APIs (e.g., OpenGL, Vulkan, Mesa)
• Experience with Linux driver development
• Familiarity with CLC/assembly compute kernels

Job Responsibilities

• Bring up, test, and debug Neural Networks using frameworks like TensorFlow, PyTorch, and ONNX for Mirafra
• Enhance and maintain TVM capabilities and features
• Optimize neural networks for efficiency and stability
• Develop and maintain Model Conversion Toolchain software stack
• Perform tasks related to:
• Network optimization
• Node fusion
• Graph lowering
• Adding custom operations
• Profiling and performance fine-tuning