ML Compiler Engineer, TPU Performance Optimizations
Bengaluru, Karnataka, India
INR 2500K-5000K (estimate) Senior-level Full Time
Tasks
- Build ML stack understanding
- Build compiler solutions for performance
- Debug performance issues in ML workloads
- Develop compiler optimizations for ML workloads
- Improve ML workload efficiency
- Provide technical leadership and mentorship
Perks/Benefits
- N/A
Skills/Tech-stack
C++ | CUDA | Compiler optimization | Computer Architecture | Debugging | GPU Programming | JAX | Kernel Fusion | Machine Learning | Performance Profiling | Performance optimization | Pipelining | Preinitetching | PyTorch | Python | Software Architecture | TPU Performance | TPU Performance Optimization | Tensor Processing Unit | Tensor processing | XLA
Education
Related jobs
-
Vice President - AI/ML Technical Developer INR 4000K-5199KAgentic Frameworks | Alerting | Automated testing | Cause analysis | Code reviewExecutive-level Full TimeBengaluru, Karnataka, India7h ago
-
Data Engineer (Python ETL) INR 800K-2000KETL | Oracle Database | PL/SQL | Performance Tuning | PythonMid-level Full TimeIN-MH-Pune7h ago
-
Mid-level Full TimeIN-MH-Pune7h ago
-
API Design | AWS | Apache Spark | Azure | Big DataSenior-level Full TimeBangalore - Karnataka - India Pune …7h ago
-
Senior Analytics Engineer INR 3000K-4000KAWS | Cloud platform | Confluent Cloud | DBT | Data IngestionAgile team collaboration | Knowledge sharing culture | Mentorship opportunitiesSenior-level Full TimeBangalore - Karnataka - India Pune …7h ago
-
Snowflake Data Engineer- Consultant INR 1500K-2000KAWS Glue | AWS Lambda | AWS S3 | Access policies | AirflowCorporate social responsibility programs | Discounts on products and services | Maternity leave | Networking opportunities | Paternity leaveMid-level Full TimeBangalore, Karnataka, India8h ago
-
Principal Engineer, Technical Architect INR 2000K-5000K.NET | API first | API-first design | Access Control | Agent OrchestrationSenior-level Full TimeRemote, India R8h ago
-
Principal Engineer, Technical Architect INR 2000K-5000K.NET | API Design | Access Control | Artificial Intelligence | Batch inferenceSenior-level Full TimeRemote, India R8h ago
-
Data Scientist - Senior INR 3600K-5000KAgile | Anomaly Detection | CI/CD | Classification | Computer VisionSenior-level Full TimePune, Maharashtra, India8h ago
-
Applied AI/ML Engineer INR 1800K-3000KAgent workflows | Agentic Systems | Auditability | Drift Detection | Error HandlingMid-level Full TimeBengaluru, Karnataka, India8h ago
-
Senior-level Full TimeIndia9h ago
-
AI Data Foundation Engineer INR 3000K-5000KApache Airflow | Apache Beam | Apache Kafka | Apache Spark | CI/CDSenior-level Full TimeIndia9h ago
-
Associate Director (AI Hub - GDC) INR 1500K-3300KAWS | Apache Kafka | CI/CD | Cassandra | Cloud platformMid-level Full TimeHyderabad, Telangana, India10h ago
-
Sr IT Analyst INR 1500K-2500KETL | Informatica IDMC | Informatica PowerCenter | Linux | Python24 7 shift model | On-call supportSenior-level Full TimeBengaluru, Karnataka, India10h ago
-
Senior-level Full TimeChennai, Tamil Nadu, India11h ago
-
Senior-level Full Timebengaluru, India11h ago
-
Senior-level Full Timebengaluru, India11h ago
-
Senior-level Full Timebengaluru, India11h ago
-
Sr.Data Engineering INR 2000K-2500KAzure Data | Azure Data Factory | Azure Databricks | Azure Event | Azure Event HubsSenior-level Full Timebengaluru, India11h ago
-
Sr.Data Engineering INR 2000K-2500KAzure Data | Azure Data Factory | Azure Databricks | Azure Event | Azure Event HubsSenior-level Full Timebengaluru, India11h ago
-
Mid-level Full Timebengaluru, India11h ago
-
AI Intern INR 180K-240KAzure | Data Preprocessing | Debugging | FastAPI | FlaskMentorship | On site work experienceEntry-level Full Time InternshipBangalore, Karnataka, India11h ago
-
Mid-level Full TimeBangalore, Karnataka, India12h ago
-
DTICI_Engineering_Data_Analysis_Engineer_T9 INR 2000K-2800KAI | Apache Spark | Centralized Logging | Data Analysis | Data WranglingSenior-level Full TimeBengaluru, Karnataka, India12h ago
-
Assistant Manager - Cloud Engineering INR 1800K-2500KApache Airflow | Apache Kafka | Apache Spark | Azure | Azure DataMid-level Full TimePune, Maharashtra, India12h ago