Research Engineer, AI Models

Posted 14 Days Ago
Hiring Remotely in India
Remote
Senior level
Artificial Intelligence • Hardware • Software
The Role
Develop and optimize AI models for efficient inference on custom silicon. Build fine-tuning pipelines, implement acceleration techniques (quantization, sparsity, distillation), create benchmarking and profiling tools, and collaborate with hardware, compiler, and quantization teams to translate algorithmic improvements into real hardware gains.
Summary Generated by Built In

Research Engineer, Applied AI 

Location: India (or Remote-friendly with travel) 

About EnCharge AI: 

EnCharge AI is building the next generation AI platform. Our novel in-memory-computing architecture delivers a 10x step-function improvement in compute energy efficiency and performance for AI inference workloads. As the demands of artificial intelligence move beyond today's models, we believe fundamental underlying infrastructure must evolve. We are an experienced team of AI researchers, silicon & systems engineers, and architects backed by leading investors, poised to become the essential platform for the next wave of AI innovation. 

The Opportunity: 

Modern AI workloads—from large language models to diffusion-based generators to multimodal systems—represent some of the most compute-intensive frontiers in AI, and some of the most promising applications for our hardware’s energy efficiency advantages. We’re building a vertically integrated AI stack that will showcase the transformative potential of our silicon while delivering real value to customers today. 

We are seeking a Research Engineer to push the boundaries of AI model capability, quality, and efficiency. You’ll build fine-tuning and post training pipelines, develop rigorous benchmarking frameworks, and work at the intersection of ML research and hardware-aware optimization—ensuring our models run beautifully on our silicon. 

This is a role for someone who thrives at the boundary between research and engineering. You’ll read papers, implement techniques, and ship production-quality code—all in service of making AI inference faster, cheaper, and better. 

Key Responsibilities: 

  • Algorithmic Acceleration: Research and implement state-of-the-art techniques to accelerate AI inference—quantization, sparsity, distillation, speculative decoding, caching strategies, and architectural modifications. Systematically characterize tradeoffs between model quality, latency, throughput, and power consumption to find optimal operating points across different use cases.  
  • Hardware Co-Design: Partner closely with hardware, compiler, and quantization teams to ensure algorithmic improvements translate to real gains on our silicon. Identify optimizations aligned with our architecture's strengths—maximizing throughput while minimizing power. Shape the feedback loop between model development and hardware. 
  • Evaluation: Build profiling tools and comprehensive benchmarking frameworks to understand compute bottlenecks, measure model quality across standard and domain-specific evals, and track efficiency metrics.  
  • Applied Research: Build robust fine-tuning workflows for modern AI models, enabling rapid experimentation with LoRA, adapters, and full fine-tuning. Stay current with the rapidly evolving landscape—evaluate new architectures, implement promising techniques, and contribute insights that inform technical and go-to-market strategy. 

Qualifications: 

  • 5+ years of experience in ML research, applied ML, or ML systems 
  • Strong fundamentals in Python and PyTorch 
  • Hands-on experience with transformers, diffusion models, state space models etc. 
  • Experience fine-tuning large models and building training/evaluation pipelines 
  • Deep understanding of transformers, attention mechanisms, & optimization techniques 
  • Comfort reading and implementing techniques from research papers 

Nice to Have: 

  • Experience with efficient inference techniques (KV cache optimization, attention variants, MoE routing, flow matching) 
  • Background in hardware-aware ML optimization or quantization 
  • Familiarity with profiling tools (PyTorch Profiler, Nsight, custom instrumentation) 
  • Publications in generative modeling, efficient inference, or ML systems 
  • Contributions to open-source ML projects 

Skills Required

  • 5+ years of experience in ML research, applied ML, or ML systems
  • Strong fundamentals in Python and PyTorch
  • Hands-on experience with modern AI models (transformers, diffusion models, or other generative architectures)
  • Experience fine-tuning large models and building training/evaluation pipelines
  • Deep understanding of transformers, attention mechanisms, and optimization techniques
  • Comfort reading and implementing techniques from research papers
  • Experience with efficient inference techniques (KV cache optimization, attention variants, MoE routing, flow matching)
  • Background in hardware-aware ML optimization or quantization
  • Familiarity with profiling tools (PyTorch Profiler, Nsight, custom instrumentation)
  • Publications in generative modeling, efficient inference, or ML systems
  • Contributions to open-source ML projects
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Santa Clara, CA
31 Employees
Year Founded: 2022

What We Do

EnCharge AI is a leader in advanced AI hardware and software systems for edge 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.

Similar Jobs

Ericsson Logo Ericsson

Head MOAI NEO EHS

Cloud • Information Technology • Internet of Things • Machine Learning • Software • Cybersecurity • Infrastructure as a Service (IaaS)
In-Office or Remote
8 Locations
88000 Employees

Capco Logo Capco

Hedge Accounting

Fintech • Professional Services • Consulting • Energy • Financial Services • Cybersecurity • Generative AI
Remote or Hybrid
India
6000 Employees

Quillbot Logo Quillbot

Systems Engineer

Artificial Intelligence • Edtech • Mobile • Natural Language Processing • Productivity • Software
Easy Apply
Remote
India
232 Employees

Capco Logo Capco

Data Analyst

Fintech • Professional Services • Consulting • Energy • Financial Services • Cybersecurity • Generative AI
Remote or Hybrid
India
6000 Employees

Similar Companies Hiring

Hanover Park Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
42 Employees
Kepler  Thumbnail
Fintech • Software
New York, New York
6 Employees
Onshore Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
60 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account