Software Engineer, Machine Learning, Edge Tensor Processing Unit, Bengaluru

Role Responsibilities: Develop and train edge-optimized models for generative AI, computer vision, natural language, and speech use cases. Co-develop optimization tools and infrastructure for customizing and training neural networks, as well as exploring architecture design spaces. Collaborate with researchers and application developers to customize neural network architectures for various applications.Write efficient machine learning training pipelines and assist in building profiling and visualization tools for model performance. Job Requirements: Bachelor's degree in Computer Science or related technical field, or equivalent practical experience. 2 years of experience with software development in C++/Python and data structures/algorithms.2 years of experience in developing and training machine learning models. Experience with deep learning frameworks such as TensorFlow, PyTorch, or JAX is preferred.