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
-
Software Engineer (C++/DS/Algorithm) INR 1716K-2800KAlgorithms | C# | C++ | Code Reviews | Data StructuresEntry-level Full TimeNoida - Uttar Pradesh - India5h ago
-
Senior-level Full TimeGurugram, Haryana, India6h ago
-
Python AI /Senior Consultant Specialist INR 3000K-5000KAWS | Azure | CI/CD | Chroma | Data AnalysisContinuous professional development | Flexible working | Inclusive diverse environmentSenior-level Full TimePune, Maharashtra, India R7h ago
-
APIs | Agent workflows | C++ | Caching | CancellationSenior-level Full TimeBengaluru, India13h ago
-
AI Engineer/Scientist INR 1800K-3000KAWS | Agentic Workflows | C++ | Data Structures | Data Structures and AlgorithmsHybrid work schedule | Mentorship | Onsite days 3 days per weekMid-level Full TimeBengaluru R16h ago
-
Lead Engineer (Embedded Platform Software) INR 2000K-3465KARM | Board Support | Board Support Package | Bootloader | C#Car lease programme | Contingency leave | Employee scholar programme | Fuel and maintenance driver wages | Group health insuranceSenior-level Full TimeIN-KA-BENGALURU-NORTHGATE ~ Sy No 2/2 Venkatala …19h ago
-
Lead Data Scientist INR 2800K-4064KAWS | AWS S3 | Amazon EC2 | Amazon EKS | AzureContinuing education program | Continuous learning resources | Family-friendly perks | Generous time off | Health care coverageSenior-level Full TimeIN - GURUGRAM DLF DOWNTOWN BLOCK …19h ago
-
Lead Software Developer INR 3715K-5449KAPIs | AWS | Agentic AI | Artificial Intelligence | AzureHybrid work environment | Work-life flexibilitySenior-level Full TimeBengaluru, India19h ago
-
C# | C++ | Debugging | Distributed Systems | File systemsMentorshipMid-level Full TimePune, Maharashtra1d ago
-
Staff AI Engineer - Servicing Enablement Platform INR 3715K-5449KAWS | Airflow | Compliance | GCP | GRPCSenior-level Full TimeBengaluru, KA, India1d ago
-
Senior Forward Deployed Applied Scientist I (Agentic AI) INR 3419K-5000KAgent systems | Asyncio | Autogen | Deep learning | Distributed TrainingConference reimbursement | Employee assistance program | Employee stock purchase program | Equity compensation | Flexible time offSenior-level Full TimeBengaluru1d ago
-
Mid-level Full TimeBangalore, Karnataka, IN1d ago
-
Senior AI Engineer INR 2500K-5000KAPI Design | Agile | Azure | Azure AI | Azure ArcHybrid work | On-call supportSenior-level Full TimeBangalore, Karnataka, IN1d ago
-
Software Engineer III - ( Java , Python) INR 2000K-3500KAI Assisted Development | AWS | Agile | CI/CD | JavaSenior-level Full TimeHyderabad, Telangana, India1d ago
-
Software Engineer II INR 1500K-2500KAzure | Azure Data | Azure Data Lake | Azure Data Lake Storage | Azure Data Lake Storage Gen2Mid-level Full TimeIndia, Telangana, Hyderabad1d ago
-
Lead Data Scientist-(VLM-Multimodal AI ,CV, Deep Learning) INR 4000K-5500KComputer Vision | Distributed Training | Embeddings | Faiss | Fine TuningCollaborative inclusive work environment | Employee wellness programs | Flexible-hybrid work | Learning and growth opportunities | Performance-based bonusSenior-level Full TimeMumbai, India1d ago
-
Sr Software Engineer (C++, Edge AI) INR 3242K-4460KBenchmarking | C++ | CPU Utilization | CUDA | Computer VisionCareer development | Continuous learning | Flexible work arrangements | Work-life balanceSenior-level Full TimeMumbai, India1d ago
-
APIs | Agentic AI | Azure | Bagging | BoostingFlexibility programmes | Inclusive benefits | Mentorship | Work-life balance supportSenior-level Full TimePune, India1d ago
-
IN_Senior Associate_AL/ML Developer _D&A_Advisory _Noida INR 2500K-4500KAI Pipelines | API Gateway | AWS | AWS Lambda | AWS Step FunctionsFlexibility programmes | Inclusive benefits | Mentorship | Wellbeing supportSenior-level Full TimeNoida, India1d ago
-
AWS | Angular | Apache Spark | Azure | C#Senior-level Full TimeIndia, Hyderabad, DVS, SEZ-1 – Orion …1d ago
-
AI/ML Engineer INR 1800K-3465KAPI Development | AWS | Asynchronous processing | Azure | Distributed ComputingAugust Free Fridays | Education support | Flexible working hours | Paid volunteer time | Well-being resourcesMid-level Full TimeHyderabad (Flexible), India1d ago
-
Software Engineering LMTS - Tableau INR 3150K-4500KAWS | Agile methodologies | Azure | Build Automation | C#Executive coaching | On-demand training | Volunteer opportunitiesSenior-level Full TimeIndia - Hyderabad1d ago
-
Senior AI Engineer INR 3264K-4800K.NET | AI Agent | API | AWS | Advanced SecurityCultural celebrations | Financial wellness programs | Healthcare benefits | Hybrid work flexibilitySenior-level Full TimeIndia Hyderabad (Galaxy)1d ago
-
Senior-level Full TimeBangalore, In, India1d ago
-
Senior AI Engineer INR 3000K-4200KAsync Python | Asyncio | Embedding Models | Git | LangchainAccidental coverage | Business casual dress code | Employee assistance program | Flexible personal time off | Hands-on team trainingSenior-level Full TimeHyderabad India - Block B1d ago