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
-
AWS | Adversarial Machine Learning | Amazon SageMaker | Anonymization | AzureCutting-edge AI security work | Flexible working hours | Fully remote | Global cross-functional collaboration | Opportunity to shape AI security best practicesSenior-level Full TimeIndia R7h ago
-
A/B | A/B Testing | AWS | Azure | B testingFlexible working hours | Flextime policy | Hybrid work model | International collaboration | Technical growth opportunitiesSenior-level Full TimeIndia7h ago
-
Principal Engineer, Storage Data Science and Analytics INR 3200K-4590KAIOps | Anomaly Detection | Apache Spark | C# | C++Career Development Programs | Flexible work arrangements | Health and wellbeing benefitsSenior-level Full TimeBengaluru, Karnātaka, India11h ago
-
Principal Engineer, Storage Data Science and Analytics INR 3000K-3900KAIOps | Agentic Workflows | Anomaly Detection | C# | C++Senior-level Full TimeBengaluru, Karnātaka, India11h ago
-
Software Engineer- Embedded Developer- C/C++ INR 2000K-3500KArtificial Intelligence | Board Support | Board Support Package | C# | C++Cafeteria | Healthcare | On-site gym | Philanthropy | Social interest groupsSenior-level Full TimeIND-BANGALORE, India11h ago
-
AI/ML engineer - Associate Research Scientist INR 1500K-2500KAIOps | API Development | AWS | Anomaly Detection | AutomationContinuous learning and professional development | Employee Assistance Program (EAP) | Flexible working environment | LinkedIn Learning | Volunteer time offMid-level Full TimeChennai, TN, India18h ago
-
Mid-level Full TimeHadapsar, Pune, Maharashtra, IND, 41101318h ago
-
Senior-level Full TimeIN-MH-Pune23h ago
-
Snr AI INR 2500K-5000KAI Search | Amazon SageMaker | Azure AI | Azure AI Search | Azure Machine LearningSenior-level Full TimeIN-HR-Gurgaon23h ago
-
API Design | AWS Bedrock | AWS Lambda | Amazon Web Services | AzureEmployee wellness programs | Flexible work arrangements | Health insurance | Inclusive global work culture | Learning and certification supportSenior-level Full TimeIndia1d ago
-
Software Engineer III - Python Developer + AWS + LLM / Gen AI INR 3500K-5000KAWS | Agile | Azure | CI/CD | Data PipelinesSenior-level Full TimeBengaluru, Karnataka, India1d ago
-
API Development | AWS | Algorithms | Azure | Backend DevelopmentAccess to development resources | Autonomy | Career growth | High ownership | Hybrid workMid-level Full TimeIndia1d ago
-
API Design | AWS | Azure | CI/CD | DockerAgile collaboration | Continuous learning | Fully remote | Leadership opportunities | Professional developmentSenior-level Full TimeIndia R1d ago
-
Sr Software Engineer (Scala /Java and Python) INR 3700K-5000KBenchmarking | Computer Vision | Data Pipelines | Distributed Systems | EmbeddingsCross-functional team projects | Learning Campus training | Personal development supportSenior-level Full TimeMumbai, India1d ago
-
Sr Software Engineer (C++, AI) INR 2500K-4460KBenchmarking | C++ | CPU Utilization | CUDA | Computer VisionCareer development | Continuous learning | Flexible work arrangements | Work-life balanceSenior-level Full TimeMumbai, India1d ago
-
Junior AI Developer INR 180K-192KClassification | Clustering | ETL | Excel | GitCareer guidance | Hands-on experience | Mentorship | Work on live projectsEntry-level Full TimeDGS India - Bengaluru - Manyata …1d ago
-
Senior-level Full TimeIN-KA-BENGALURU-NORTHGATE ~ Sy No 2/2 Venkatala …1d ago
-
Associate Software Engineer - Data Scientist INR 1500K-2500KApache Spark | Azure Blob | Azure Blob Storage | Azure Cognitive | Azure Cognitive ServicesAgile collaboration | Certification sponsorship | Continuous learning budget | Hybrid work model | MentorshipMid-level Full TimeJohnson Controls India COEE11d ago
-
Senior-level Full TimeChennai - Vikram Sarabhai Instronic Estate, …1d ago
-
Senior-level Full TimeGurgaon, India1d ago
-
AWS | Alerting | Autogen | Data Ingestion | Data PreprocessingSenior-level Full TimeBangalore - Carina, India R1d ago
-
IN_Specialist 3_GenAI/Python_AI in Delivery_Advisory_Gurgaon INR 1800K-2700KAWS | CI/CD | Cloud platform | Data Augmentation | Data PreprocessingFlexibility programs | Inclusive benefits | Mentorship | Wellbeing supportSenior-level Full TimeGurugram Novus Tower, India1d ago
-
Lead AI Platform & Automation Engineer - GCP, Vertex AI, IBM Watsonx, MLFlow, Terraform INR 2500K-5000KC++ | CI/CD | Cloud Security | Cloud platform | Computer VisionSenior-level Full TimeIN - TDC 1 (IN110), India1d ago
-
AI/ML Engineer INR 3799K-5000KAPI Development | AWS | Agentic AI | Alerting | AzureCareer development opportunities | Comprehensive benefits | Inclusion cultureMid-level Full TimePrimary location: Bengaluru, Karnataka1d ago
-
AI/ML Engineer INR 3799K-5000KAPIs | Amazon SageMaker | Anomaly Detection | Azure ML | CI/CDCareer development opportunities | Comprehensive benefits | Inclusion-focused cultureMid-level Full TimePrimary location: Bengaluru, Karnataka1d ago