company logo

Software Engineer- Machine Learning + TVM

Hyderabad
Full-Time
Mid-Level: 4 to 10 years
1.2L - 1.5L (Per Year)
Posted on Jul 18 2025

About the Job

Skills

TensorFlow
PyTorch
ONNX
TVM
Tensor Virtual Machine
Python
NumPy
Object-Oriented Programming (OOP)

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


About the company

ATEM is a Strategic Provider of Software Solutions, Consulting services & BPO across Industry verticals. We are a 20-Year-old organization established in North America and in India. ATEM provides full life cycle software development and maintenance, contractual and temporary staffing and complete BPO Solutions in multiple industry verticals. We specialize in temp hiring & payroll processing. We pr ...Show More

Industry

IT Services and IT Consul...

Company Size

201-500 Employees

Headquarter

Chetpet, Tamil Nadu