AI Software Lead – PyTorch & CUDA Runtime (Next-Gen Accelerator) 10+ years
Bengaluru, KA, India
INR 3242K-4200K (estimate) Senior-level Full Time
Tasks
- Analyze and resolve performance bottlenecks
- Collaborate with compiler and low level software teams
- Design neural network operators
- Develop neural network ops with CUDA custom runtime APIs
- Implement runtime level optimizations
- Optimize neural network operators
- Optimize operator fusion
- Own runtime to neural network layer interfaces
Perks/Benefits
- N/A
Skills/Tech-stack
C plus plus | C# | CUDA | Execution Engine | Graph optimization | JAX | Kernel development | Memory hierarchy | Operator Kernel Development | Operator fusion | Parallel execution | PyTorch | Scheduling | TensorFlow
Education
N/A
Roles
Related jobs
-
ARM Microcontroller | Arc | C# | Clock | DMAEntry-level Full TimeBengaluru, KA, India16h ago
-
API Design | AWS | Apache Spark | Azure | Big DataSenior-level Full TimeBangalore - Karnataka - India Pune …16h ago
-
Principal Engineer, Technical Architect INR 2000K-5000K.NET | API Design | Access Control | Artificial Intelligence | Batch inferenceSenior-level Full TimeRemote, India R17h ago
-
ML Compiler Engineer, TPU Performance Optimizations INR 2500K-5000KC++ | CUDA | Compiler optimization | Computer Architecture | DebuggingSenior-level Full TimeBengaluru, Karnataka, India18h ago
-
Associate Director (AI Hub - GDC) INR 1500K-3300KAWS | Apache Kafka | CI/CD | Cassandra | Cloud platformMid-level Full TimeHyderabad, Telangana, India19h ago
-
Senior-level Full Timebengaluru, India20h ago
-
Senior-level Full Timebengaluru, India20h ago
-
Senior-level Full Timebengaluru, India20h ago
-
Mid-level Full Timebengaluru, India20h ago
-
Mid-level Full TimeBangalore, Karnataka, India21h ago
-
Senior-level Full TimeIndia, Karnataka, Bangalore23h ago
-
Principal Engineer, Sitecore Architect (Optimizely) INR 4000K-6500KA/B | A/B Testing | A/B Testing Tools | ASP.NET MVC | ASP.Net CoreSenior-level Full TimeRemote, India R1d ago
-
AI Model Routing | AI model | AI platforms | AWS | Audit LoggingCareer development | Collaborative culture | Generous time off | Mental health support | Remote-firstSenior-level Full TimeIndia1d ago
-
Embedded System Senior Test Engineer INR 1445K-2156KBluetooth | C# | C++ | Contact Lab | Contactless Smart CardSenior-level Full TimeBangalore, India1d ago
-
AWS Bedrock | Amazon Aurora | Amazon DynamoDB | Amazon RDS | Amazon SageMakerSenior-level Full TimeNoida, India1d ago
-
Staff Engineer - NSP Embedded System Software INR 2000K-3000KARM architecture | Android | Assembly Language | Build System | C#Senior-level Full TimeBangalore, Karnataka, India1d ago
-
Senior Applied Research Engineer, Accelerator Algorithms INR 3750K-6000KBenchmarking | Bottleneck analysis | C++ | CUDA | Co-designSenior-level Full TimeIndia, Bengaluru1d ago
-
Lead AI ML Engineer INR 3000K-5000KAWS | Agents | Agile | Artificial Intelligence | AzureCareer development opportunities | Comprehensive benefitsSenior-level Full TimePrimary location: Noida, India1d ago
-
Senior-level Full TimeBengaluru, Karnataka1d ago
-
Mid-level Full TimeHyderabad, India1d ago
-
Senior AI Engineer - GenAI, AI Platform & Voice AI INR 2000K-5000KAPI Development | Airflow | C# | C++ | DagsterSenior-level Full TimeBangalore, India1d ago
-
Gen AI Engineer Intermediate Analyst(4-7Years) INR 1200K-2800KAWS | Agent architecture | Azure | Cloud platform | ContainersMid-level Full TimeChennai, Tamil Nadu, INDIA; Bengaluru, Karnataka, …1d ago
-
Senior-level Full TimeBengaluru, KA, India1d ago
-
Algorithm Optimization | C# | C++ | Data Structures | Digital LogicHealth and wellness benefits | Non monetary benefitsSenior-level Full TimeBengaluru, India1d ago
-
R&D Engineering, Principal Engineer (EDA Embedded Firmware FPGA) INR 2500K-4460KAlgorithms | C# | C++ | Data Structures | Graph AlgorithmsEmployee support for learning and growth | Health and wellness benefitsSenior-level Full TimeNoida, India1d ago