📍 Jobs Near Me
📍

HiringNearMe.work

Local Jobs, Zero Commute

📍 Local Job Near You

pytorch automation

🏢
LTM
📍 Bengaluru, India
📍
Location Bengaluru
📅
Posted June 03, 2026
🚗
Commute Local Area
🎯
Local Opportunity Near You!

This job is in your area. Enjoy a short commute and work close to home.

📋
Job Description

Hiring Pytorch Automation specialist, with expertise in Pytorch workloads and triton.


  • Develop PyTorch workloads that stress model‑level execution (e.g., large GEMMs, attention patterns, MoE‑like behavior, mixed precision, long‑running loops)
  • Author custom Triton kernels to directly stress hardware execution units, memory hierarchies, and synchronization paths
  • Build parameterized stress harnesses that can scale with problem size, number of devices, and runtime duration
  • Integrate workloads with existing tooling for profiling, monitoring, and failure triage
  • Collaborate with platform, firmware, and SDK teams to ensure workloads target known risk areas and emerging issues
  • Document usage patterns and provide reproducible scripts for lab and CI usage

Expected Deliverables

  • A library of reusable PyTorc...

Apply for This Job

Submit Application

Quick and secure application process

📍 Location Details

🌆
City
Bengaluru
🗺️
Country
India
🚗
Commute
Local Area

🔍 More Jobs Nearby

Explore other opportunities in Bengaluru

View Local Jobs