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
-
Mid-level Full TimePune, Maharashtra, India18h ago
-
AES - DE - Generative AI Application Developers INR 2500K-5000K.NET | AI Agents | AI Foundry | AI guardrails | ASP.Net CoreSenior-level Full TimePune, Maharashtra, India19h ago
-
Mid-level Full TimeIndia, Telangana, Hyderabad22h ago
-
ML Engineer INR 2400K-3000KClassification | Clustering | Computer Vision | Deep learning | DevOpsAnnual leave | Digital learning platforms | Employee assistance program | Flexible working arrangements | Learning & development opportunitiesMid-level Full TimeBangalore, IN23h ago
-
Principal Software Engineer INR 2500K-5000KAI orchestration | Artificial Intelligence | By Design | C# | C++Senior-level Full TimeIndia, Telangana, Hyderabad1d ago
-
Lead Data Scientist INR 3584K-5000KAgent Orchestration | Agent systems | Asynchronous task queues | CI/CD | Cloud Object StorageHealth insurance | Paid time offSenior-level Full TimeIN - HYDERABAD SKYVIEW, India1d ago
-
MTS-1 Machine Learning Engineer INR 1500K-2500KAWS | Apache Flink | Apache Hadoop | Apache Spark | AzureMid-level Full TimeBangalore, India1d ago
-
MTS-2 Machine Learning Engineer INR 3600K-5400KAWS | Apache Flink | Apache Hadoop | Apache Spark | Big DataSenior-level Full TimeBangalore, India1d ago
-
Senior Machine Learning Engineer, ITC INR 3000K-5000KAPI Gateway | AWS | Airflow | Algorithms | CI/CDAgile working environment | Distributed Global Team | Professional growth learning | Team mentorshipSenior-level Full TimeINDIA TECHNOLOGY CENTER, India1d ago
-
LLM Data Scientist, AVP INR 3000K-6000KAWS | Anomaly Detection | Azure | CI/CD | DatabricksEmployee networks | Flexible work/life support | Inclusive development opportunities | Paid volunteer days | Vibrant employee networksExecutive-level Full TimeBangalore, India1d ago
-
Data Scientist- Gen AI INR 1800K-2800KAWS Bedrock | Audio analysis | Azure | Azure OpenAI | Cloud platformMid-level Full TimeBengaluru, India1d ago
-
SWE-3 Machine Learning Engineer INR 1500K-2500KAWS | Apache Flink | Apache Hadoop | Apache Spark | AzureMid-level Full TimeBangalore, India1d ago
-
Sr MTS Machine Learning Engineer INR 3715K-5449KA/B | A/B Testing | Alerting | Apache Flink | Apache HadoopSenior-level Full TimeBangalore, India1d ago
-
Data Scientist f Submission Data and Content Generation & Reuse (AIDCG) - Pharma R&D INR 3000K-5000KAI Governance | API | AWS AgentCore | Agentic RAG | Artificial IntelligenceSenior-level Full TimeHyderabad RDT, India1d ago
-
Sr Software Engineer III - AI (IND) INR 2500K-4500KA/B | A/B Testing | API Management | AWS | Apache SparkSenior-level Full TimeIND-KA-Bengaluru, Bagmane Solarium City, India1d ago
-
Lead Data Scientist INR 3584K-5000KAsynchronous Distributed Systems | CI/CD | Cloud infrastructure | Containerization | Deep learningFlexible time off | Health care coverageSenior-level Full TimeIN - HYDERABAD SKYVIEW, India1d ago
-
AWS | Adversarial Networks | Agile | Computer Vision | Diffusion ModelsHealth and medical insuranceEntry-level Full TimeIND-KA-Bengaluru, Bagmane Solarium City, India1d ago
-
Senior-level ContractShaikpet, India1d ago
-
Senior-level Full TimeHyderabad, India1d ago
-
Senior Software Engineer -Senior AI Engineer INR 1492K-2500KApp Service | Azure App | Azure App Service | C# | C++Senior-level Full TimeIndia, Karnataka, Bangalore; India, Telangana, Hyderabad1d ago
-
Advanced Engineer-AI/ML INR 2500K-4680KAI orchestration | API Integration | ASR | AWS | Action modelsEmployee discounts | Training opportunities | Tuition reimbursement | Wellness benefitsSenior-level Full TimeBangalore - Karnataka, India - Kalyani …1d ago
-
Staff Software Engineer - Data Path INR 1500K-3000KC# | C++ | Concurrency | Distributed Storage | FilesystemsSenior-level Full TimePune Office1d ago
-
AI/ML Senior Specialist INR 3242K-5000KBM25 | CI/CD | Chroma | Docker | EmbeddingsHybrid work environment | On-call rotationSenior-level Full TimeKochi, Kerala, India1d ago
-
Firmware Dev with NW & Storage Software Development Eng. INR 1000K-2000KARM | AXI | C# | C++ | DDR SDRAMMid-level Full TimeIN,Hyderabad-Sattva1d ago
-
Senior Software Engineer, AI/ML, Google Cloud INR 2500K-5000KC plus plus | Data Processing | Data Structures | Data structures algorithms | DebuggingSenior-level Full TimeHyderabad, Telangana, India1d ago