AI Software Lead – PyTorch & CUDA Runtime (Next-Gen Accelerator)
Bengaluru, KA, India
INR 2475K-3380K (estimate) Senior-level Full Time
Tasks
- Collaborate with compiler teams and framework teams
- Design and optimize neural network operators
- Develop neural network ops with CUDA custom runtime APIs
- Drive runtime level optimizations
- Identify and resolve performance bottlenecks
- Implement and optimize operator fusion
- Own runtime to neural network layer interfaces
Perks/Benefits
- N/A
Skills/Tech-stack
C# | C++ | CUDA | Compiler runtime interaction | Compiler/runtime | Execution Engine | GPU Computing | Graph optimization | JAX | MLIR | Memory hierarchy | NPU Computing | Operator fusion | Parallel execution | PyTorch | Runtime Systems | TensorFlow | XLA
Education
N/A
Roles
Related jobs
-
Software Engineer - Embedded Linux INR 1236K-1940KAOSP | Android | Android HAL | C++ | C++11Employee discounts | Employee recognition rewards | Inclusive work environment | Training opportunities | Tuition reimbursementSenior-level Full TimeBangalore - Karnataka, India - Kalyani …10h ago
-
Advanced Engineer - Embedded Linux C++ INR 1236K-1940KAOSP | Android | Android HAL | C++ | C++11Employee discounts | Employee recognition | Training opportunities | Tuition reimbursement | Wellness benefitsSenior-level Full TimeBangalore - Karnataka, India - Kalyani …10h ago
-
Advanced Engineer - Embedded C++ linux INR 1236K-1940KAOSP | Android | Android HAL | C++ | C++11Employee discounts | Employee recognition program | Inclusive work environment | Training opportunities | Tuition reimbursementSenior-level Full TimeBangalore - Karnataka, India - Kalyani …10h ago
-
Information Sys Engineer INR 1000K-2000KAlgorithms | CI/CD | Cloud Computing | Computer Architecture | Data StructuresHybrid work environmentMid-level Full TimeBangalore, India Office (BANGALORE)17h ago
-
Senior-level Full TimePune17h ago
-
AWS | Apache Spark | Databricks | GenAI | Hugging FaceFlexible work arrangements | Professional growth opportunitiesSenior-level Full TimeIndia17h ago
-
Applied AI ML Senior Associate INR 2500K-4500KAPIs | AWS Bedrock | Azure OpenAI | CI/CD | DatabricksSenior-level Full TimeBengaluru, Karnataka, India18h ago
-
Software Engineer_Data Science INR 1500K-2000KAWS | Amazon SageMaker | Azure | Azure Services | ClusteringMid-level Full TimeBangalore, India Office (BANGALORE)19h ago
-
Senior-level Full TimeIndia GCC-Puppalaguda Village23h ago
-
Staff Software Engineer/Data Scientist INR 2500K-5000KA/B | A/B Testing | AI Services | AWS Bedrock | AWS SageMakerSenior-level Full TimeIndia GCC-Puppalaguda Village23h ago
-
Staff Software Engineer/Data Scientist INR 2500K-5000KA/B | A/B Testing | B testing | Bias Mitigation | CI/CDSenior-level Full TimeIndia GCC-Puppalaguda Village23h ago
-
Staff Software Engineer/Data Scientist INR 2500K-5000KA/B | A/B Testing | AI Services | AWS SageMaker | Agentic AISenior-level Full TimeIndia GCC-Puppalaguda Village23h ago
-
Staff Software Engineer/Data Scientist INR 2500K-5000KA/B | A/B Testing | AI Services | AWS Bedrock | AWS SageMakerSenior-level Full TimeIndia GCC-Puppalaguda Village23h ago
-
Artificial Intelligence / Machine Learning Engineer Lead (GenAI-powered applications) INR 2475K-3500KAWS | Agentic AI | Agents | Azure | CI/CDSenior-level Full TimeIND BNGL FL2-3 TWR 3, India23h ago
-
Senior Software Engineer/AI-ML INR 2500K-5000K12-Factor | 12-Factor App | AWS SageMaker | Asyncio | BedrockSenior-level Full TimeIndia GCC-Puppalaguda Village23h ago
-
Senior Software Engineer/AI-ML INR 2500K-5000KAWS SageMaker | Agentic AI | Asyncio | Bedrock | CrewAISenior-level Full TimeIndia GCC-Puppalaguda Village23h ago
-
Software Engineer/AI-ML INR 2500K-5000KAI Search | AWS | AWS Bedrock | AWS SageMaker | Agent systemsSenior-level Full TimeIndia GCC-Puppalaguda Village23h ago
-
Machine Learning Lead Analyst - HIH - Evernorth INR 2500K-4500KAPI Integration | AWS | Algorithms | C# | Cloud platformFull-time employment | Remote work flexibilitySenior-level Full TimeHIH - Hyderabad, India R23h ago
-
AI and Machine Learning Engineer INR 2500K-4500KCalculus | Convolutional Neural Networks | Data Engineering | Data Science | Decision TreesFlexible work arrangements | Health insurance | Paid time off | Professional developmentSenior-level Full TimeBengaluru, Karnātaka, India23h ago
-
Software Engineer - SASE INR 1200K-2000KApplication Classification | C# | C++ | DNS | Deep Packet InspectionCareer Development Programs | Health and wellbeing benefits | Hybrid work arrangementMid-level Full TimeBengaluru, Karnātaka, India23h ago
-
Mid-level Full TimeBengaluru, BDC7C, India23h ago
-
Mid-level Full TimeHyderabad, HDC3B, India23h ago
-
Mid-level Full TimeChennai, CDC2F, India23h ago
-
IN_Senior Associate_AI/ML Developer_MS Engg_Advisory_Kolkata INR 2000K-4500KAPI Management | Acceptance Test | Acceptance Test-Driven Development | App Service | Artificial IntelligenceFlexibility programmes | Inclusive benefits | MentorshipSenior-level Full TimeKolkata DN 57, India23h ago
-
IN_Associate_AI Engineer_Data Analytics_Advisory_Bangalore INR 2000K-4800KA/B | A/B Testing | Apache Spark | B testing | CI/CDFlexibility programmes | Inclusive benefits | Mentorship | Wellbeing supportSenior-level Full TimeBengaluru Millenia, India23h ago