AI Kernel Development Lead: Optimizing Performance for Machine Learning

6 days ago

Bengaluru, Karnataka, India Tenstorrent Full time
About the Role
Tenstorrent is a leading innovator in cutting-edge AI technology, revolutionizing performance expectations, ease of use, and cost efficiency. As a pioneer in this field, we are pushing the boundaries of what is possible with software models, compilers, platforms, networking, and semiconductors.

We are growing our team of talented technologists who have developed a high-performance RISC-V CPU from scratch. Our passion for AI and commitment to building the best AI platform possible drives us to collaborate, be curious, and solve hard problems. We value these qualities and are seeking contributors of all seniorities to join our team.

As the AI Kernel Developer Lead at Tenstorrent, you will play a pivotal role in guiding the optimization of low-level workloads, kernel development, and enhancing the performance of our software for machine learning applications. You will lead a team of skilled engineers, ensuring our software operates at peak efficiency and delivers high-quality results to our clients and users.

This role is based out of Bangalore, India, and offers an exciting opportunity to work on cutting-edge AI technology.

Key Responsibilities
- **Kernel Development Leadership**: Lead the design, development, and maintenance of kernel-level software components for our applications.
- Develop and optimize kernels and kernel libraries for efficient machine learning and HPC applications.
- **Tensor Compute and Data Movement Optimizations**: Lead the implementation and optimization of tensor compute and tensor data movement kernels.
- Drive efforts in analyzing and optimizing low-level code to improve the performance and efficiency of our software, with a strong emphasis on tensor optimization.
- Collaborate with machine learning engineers and data scientists to integrate optimized kernels and low-level routines into machine learning frameworks and pipelines.
- **Performance Profiling and Strategy Development**: Identify performance bottlenecks, conduct performance profiling, and develop strategies to address and resolve them.
- Oversee the creation of comprehensive unit tests, conduct thorough debugging, and ensure the stability and reliability of kernel-level code.
- Create and maintain clear and concise documentation for code, APIs, and best practices to facilitate team collaboration.
- Stay abreast of the latest developments in kernel development, tensor optimization, and machine learning to propose and implement innovative solutions and improvements.

Requirements and Qualifications
- Bachelor's degree in Computer Science, Software Engineering, or a related field.
- Proven experience in kernel development, with a strong focus on low-level optimizations and tensor optimization.
- Proficiency in C/C++ programming languages.
- Familiarity with machine learning frameworks and concepts.
- Strong problem-solving skills and the ability to analyze and debug complex issues.
- Experience with performance profiling and optimization tools.
- Excellent communication and leadership skills.
- Self-motivated, detail-oriented, and able to work independently as well as lead a team.
- Experience with GPU programming (CUDA, OpenCL) is a plus.
- Knowledge of operating system internals is a plus.

Benefits and Growth Opportunities
The estimated salary for this role is around ₹25-30 lakhs per annum, depending on experience and qualifications.

  • AI Kernel Optimization Lead

    4 weeks ago

    Bengaluru, Karnataka, India Tenstorrent Full time

    Tenstorrent is revolutionizing the AI technology landscape with cutting-edge innovations, setting new standards for performance, ease of use, and cost efficiency.In this pivotal role, you will lead a team of skilled engineers to optimize low-level workloads, kernel development, and enhance software performance for machine learning applications. Your...

  • Senior AI Kernel Optimization Specialist

    3 weeks ago

    Bengaluru, Karnataka, India Tenstorrent Full time

    Tenstorrent is a pioneering force in cutting-edge AI technology, redefining performance expectations, ease of use, and cost efficiency. Our diverse team of technologists has developed a high-performance RISC-V CPU from scratch, and share a passion for AI and a deep desire to build the best AI platform possible.Job Overview:We are looking for an experienced...

  • Accelerator Kernel Optimization Specialist

    3 weeks ago

    Bengaluru, Karnataka, India Tenstorrent Full time

    Job Title: Accelerator Kernel Optimization SpecialistLocation: Bangalore, IndiaAbout TenstorrentTenstorrent is a pioneer in cutting-edge AI technology, pushing the boundaries of performance expectations, ease of use, and cost efficiency.Job SummaryWe are seeking an experienced Accelerator Kernel Optimization Specialist to lead our team in optimizing...

  • Machine Learning Acceleration Lead

    3 weeks ago

    Bengaluru, Karnataka, India Tenstorrent Full time

    Job Title: Machine Learning Acceleration LeadLocation: Bangalore, IndiaAbout TenstorrentTenstorrent is at the forefront of cutting-edge AI technology, driving innovation in performance expectations, ease of use, and cost efficiency.Job SummaryWe are seeking an experienced Machine Learning Acceleration Lead to guide our team in optimizing low-level workloads,...

  • AI Performance Engineering Lead

    3 weeks ago

    Bengaluru, Karnataka, India Tenstorrent Full time

    Tenstorrent is a leading innovator in cutting-edge AI technology, pushing the boundaries of performance expectations, ease of use, and cost efficiency. Our team of expert technologists has successfully developed a high-performance RISC-V CPU from scratch, driven by a shared passion for AI and a commitment to building the best AI platform possible.Job...

  • AI Software Performance Engineer

    3 weeks ago

    Bengaluru, Karnataka, India Tenstorrent Full time

    Job Title: AI Software Performance EngineerLocation: Bangalore, IndiaAbout TenstorrentTenstorrent is revolutionizing the industry with its cutting-edge AI technology, focusing on performance expectations, ease of use, and cost efficiency.Job SummaryWe are looking for an experienced AI Software Performance Engineer to lead our team in optimizing low-level...

  • Software Development Expert for Linux Kernel

    7 days ago

    Bengaluru, Karnataka, India Linux Kernel & LDD Full time

    At Linux Kernel & LDD, we're seeking a highly skilled Software Development Expert for Linux Kernel to join our team in Bengaluru. This is an exciting opportunity to work on real-world projects and gain hands-on experience in:Key Responsibilities:Developing device drivers and embedded systems using Linux kernel fundamentalsImplementing practical solutions for...

  • Machine Learning Engineer

    3 weeks ago

    Bengaluru, Karnataka, India JTek Software Solutions Pvt Ltd Full time

    Role Overview: We are seeking a skilled Artificial Intelligence Specialist to join our team at JTek Software Solutions Pvt Ltd. The ideal candidate will have a strong background in artificial intelligence, machine learning, and software development, with expertise in Python or C++ and a solid understanding of Linux systems. This role offers an opportunity to...

  • AI and Machine Learning Architect

    3 weeks ago

    Bengaluru, Karnataka, India Squareroot Consulting Pvt. Ltd. Full time

    Lead AI and Machine Learning Model DeveloperSquareroot Consulting Pvt. Ltd. is seeking an experienced AI/ML Lead and Architect to join our team in Bangalore. As a seasoned professional, you will be responsible for leading the development and architecture of machine learning models and AI systems to address specific business needs.The ideal candidate will...

  • AI and Machine Learning Expert

    3 weeks ago

    Bengaluru, Karnataka, India Skellam AI Full time

    We are looking for an experienced Data Analyst/Scientist who can leverage their skills in machine learning and deep learning to drive business growth at Skellam AI Pvt Ltd.The successful candidate will have strong analytical skills, excellent communication abilities, and a proactive mindset.Main Responsibilities:Collaborate with cross-functional teams to...

  • Machine Learning Compiler Specialist

    1 month ago

    Bengaluru, Karnataka, India Skywaves Rise Full time

    Job Title: Machine Learning Compiler SpecialistAbout the Role:At Skywaves Rise, we're seeking a skilled Machine Learning Compiler Specialist to join our team. As a key member of our AI/ML team, you'll be responsible for developing and supporting new AI/ML compiler features and technologies to accelerate deep learning models. Your expertise will enable us to...

  • AI Development Lead

    3 weeks ago

    Bengaluru, Karnataka, India Infosys Full time

    Excel in a Data Scientist role at Infosys, where you will lead the development of AI solutions and contribute to various projects.As an AI Development Lead, your primary responsibilities will include anchoring machine learning development tracks in client projects. You will be responsible for data collection, profiling, and preparation as well as AI model...

  • Machine Learning Engineer

    2 weeks ago

    Bengaluru, Karnataka, India JTek Software Solutions Pvt Ltd Full time

    Job Overview:We are looking for a highly motivated and talented Machine Learning Engineer to join our team at JTek Software Solutions Pvt Ltd. The ideal candidate will have hands-on experience in designing, developing, and deploying AI/ML models, as well as a strong understanding of machine learning frameworks such as TensorFlow, PyTorch, or Scikit-learn.Key...

  • Machine Learning Engineer for Intelligent Systems

    1 day ago

    Bengaluru, Karnataka, India Skill Secure AI Full time

    About UsSkill Secure AI is a forward-thinking company focused on innovation in the field of artificial intelligence and machine learning.We leverage cutting-edge technologies to drive data-driven solutions across various industries, continuously expanding our expertise.Job DescriptionWe are seeking a highly motivated Machine Learning Engineer to contribute...

  • AI Developer with Machine Learning Expertise

    2 weeks ago

    Bengaluru, Karnataka, India Market Cloud Ltd Full time

    Job Title: AI Developer with Machine Learning ExpertiseWe are seeking a highly skilled AI developer to join our team at Market Cloud Ltd in London. As an onsite opportunity, you will have the chance to work on cutting-edge machine learning projects and collaborate with our experienced data scientists and engineers.About the RoleThe ideal candidate will have...

  • AI Machine Learning Engineer

    1 month ago

    Bengaluru, Karnataka, India Anumana Full time

    **Job Overview**Anumana, a leading AI-powered healthcare company, is seeking an experienced Data Scientist to join our team. With a strong background in machine learning and deep learning, the ideal candidate will have a minimum of 1-2 years of experience in training and evaluating models using PyTorch and TensorFlow.The successful candidate will be...

  • Senior AI Developer

    2 weeks ago

    Bengaluru, Karnataka, India JTek Software Solutions Pvt Ltd Full time

    Job Title: Senior AI DeveloperWe are seeking a highly skilled and experienced AI/ML developer to join our team at JTek Software Solutions Pvt Ltd. The ideal candidate will have a strong background in artificial intelligence, machine learning, and software development, with expertise in Python or C++ and a solid understanding of Linux systems. This role...

  • Visionary AI and Machine Learning Expert

    3 weeks ago

    Bengaluru, Karnataka, India Thoucentric Technology Pvt. Ltd. Full time

    Job Title: AI and Machine Learning ExpertWe are seeking an exceptional AI and machine learning expert to lead the development of our computer vision capabilities.This role involves designing and implementing advanced machine learning models to drive business outcomes and improve operational efficiency.Responsibilities:Develop and train machine learning...

  • Data Scientist and Machine Learning Lead

    3 weeks ago

    Bengaluru, Karnataka, India Tata Consultancy Services Full time

    Tata Consultancy Services is a leading IT services company that provides innovative solutions to its clients across various industries.Our team of experts is seeking a highly skilled Data Scientist and Machine Learning Lead to join our organization.Salary: ₹1500000 - ₹3000000 per annumWe are looking for a seasoned professional with 4+ years of experience...

  • Machine Learning Architect Lead

    3 weeks ago

    Bengaluru, Karnataka, India Coders Brain Technology Private Limited Full time

    Coders Brain Technology Private Limited is seeking an experienced Machine Learning Architect to lead the design and development of cutting-edge AI and machine learning solutions.Job Title: ML Architect (AI Generative, Azure)We are looking for a skilled professional with expertise in AI Generative models, such as GPT and DALL-E. The ideal candidate should...