EnCharge AI Logo

EnCharge AI

AI Compiler Engineer

Posted 21 Days Ago
Be an Early Applicant
Remote
Hiring Remotely in India
Mid level
Remote
Hiring Remotely in India
Mid level
Design and implement graph compiler optimizations (fusion, quantization, tiling) to convert TensorFlow/PyTorch models into IR, improve performance and hardware utilization, collaborate with ML researchers and hardware architects, and lead a team focused on compiler and code-generation improvements.
The summary above was generated by AI

EnCharge AI is a leader in advanced AI hardware and software systems for edge-to-cloud computing. EnCharge’s robust and scalable next-generation in-memory computing technology provides orders-of-magnitude higher compute efficiency and density compared to today’s best-in-class solutions. The high-performance architecture is coupled with seamless software integration and will enable the immense potential of AI to be accessible in power, energy, and space constrained applications. EnCharge AI launched in 2022 and is led by veteran technologists with backgrounds in semiconductor design and AI systems.

About the Role

EnCharge AI is seeking a highly skilled and experienced AI Compiler Engineer to spearhead the efforts in developing and optimizing graph compilers tailored to cutting-edge AI and ML workloads. You will collaborate with hardware architects, and AI researchers to enhance performance, optimize computation graphs, and enable efficient model deployment on EnCharge’s Inference Accelerators. 

Responsibilities

Architect, design, and implement optimizations for AI model execution on graph compilers to improve performance, reduce latency, and maximize hardware utilization.

  • Work closely with ML researchers, hardware engineers, and software developers to design and deploy AI models, understanding and addressing hardware-specific challenges.
  • Work on performance optimizations for neural network models, such as layer fusion, operator fusion, and graph-level transformations.
  • Develop compiler optimizations and passes that convert high-level AI models (e.g., from TensorFlow, PyTorch) into intermediate representations (IR).
  • Implement parsing, semantic analysis, and IR generation for deep learning frameworks.
  • Research and integrate the latest advancements in compiler design, ML model optimizations, and hardware acceleration into graph compilers.
  • Provide leadership, mentorship, and technical guidance to a team of engineers focused on graph compiler optimizations. 

Qualifications

  • Bachelor’s or Master’s degree in Computer Science, Electrical Engineering, or related field (Ph.D. preferred).
  • 3+ years in compiler development, with a strong focus on AI or ML graph compilers.
  • Proficiency in AI graph compiler frameworks (e.g., MLIR, Torch-FX)
  • Solid background in hardware architectures (e.g., GPUs, TPUs, ASICs) and optimization techniques such as fusion, quantization, and tiling.
  • Familiarity with neural networks operators and code generation.
  • Strong understanding of intermediate representations, code parsing, and semantic analysis in compiler design.
  • Proficiency in C++, Python, or other programming languages commonly used in compiler development.
  • Open-source contributions to AI software frameworks and libraries is a plus
  • Demonstrated experience leading and mentoring engineering teams with successful project delivery. 

EnchargeAI is an equal employment opportunity employer in the United States.


Similar Jobs

2 Hours Ago
Remote or Hybrid
India
Junior
Junior
Fintech • Legal Tech • Software • Financial Services • Cybersecurity • Data Privacy
Manage client onboarding and CDD/KYC reviews, coordinate with onshore teams, handle client queries, track missing requirements, escalate complex issues, support high-volume onboarding projects, coach team members, and contribute to process improvements while ensuring regulatory compliance.
3 Hours Ago
Remote
Gujarat, IND
Entry level
Entry level
Artificial Intelligence • Hardware • Information Technology • Machine Learning
Assist in projects related to MES applications, optimize system setups, provide training, implement software improvements, and maintain business process integrity.
Top Skills: Enterprise Resource Planning (Erp)Micron Execution System (Mes)Product Lifecycle Management (Plm)
3 Hours Ago
Remote
Gujarat, IND
Internship
Internship
Artificial Intelligence • Hardware • Information Technology • Machine Learning
Intern will develop and prototype AI/ML models (yield prediction, anomaly detection, predictive maintenance, process optimization), extract and analyze MES, equipment, and yield data, build Python tools and Power BI dashboards, collaborate with PDE and manufacturing teams to validate models, and document and present technical findings and deployment recommendations.
Top Skills: MatplotlibMesNumpyPandasPower BIPythonPyTorchScikit-LearnSeabornSQLTensorFlow

What you need to know about the Pune Tech Scene

Once a far-out concept, AI is now a tangible force reshaping industries and economies worldwide. While its adoption will automate some roles, AI has created more jobs than it has displaced, with an expected 97 million new roles to be created in the coming years. This is especially true in cities like Pune, which is emerging as a hub for companies eager to leverage this technology to develop solutions that simplify and improve lives in sectors such as education, healthcare, finance, e-commerce and more.

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account