Software Engineer, Machine Learning, Edge Tensor Processing Unit

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.