[Japan] Machine Learning Engineer

Introduction

We are looking for a passionate and skilled Machine Learning Engineer to join our SAKURA AI Chip team at EdgeCortix. In this role, you will work at the intersection of hardware and AI, helping to accelerate the deployment of state-of-the-art neural networks on our proprietary edge AI platform. This is an exciting opportunity to contribute to groundbreaking innovation in AI hardware-software co-design and help shape the future of intelligent computing at the edge.

About Edgecortix

At Edgecortix we are a deep-tech startup revolutionizing edge computing with artificial intelligence and novel high efficiency silicon on chip design. Originating from multiple years of research, our unique AI hardware & software co-design principle and the Dynamic Neural Accelerator ® AI processor IP are geared towards positively disrupting the rapidly growing artificial intelligence edge hardware space and bring the power of AI and machine learning to all kinds of devices. Our operations are headquartered in Tokyo, Japan, with offices in Singapore, Virginia, and California in the US.

The Team

As an engineering driven company we are working to define and solve the hardest problems in AI including computer vision, speech, and natural language, geared towards real-time capabilities on small to medium form factor devices. We originated out of multiple years of research, as such at our core we value learning, intellectual curiosity, and self-starters. We have the ambitious goal of enabling cloud-level performance with significantly better energy-efficiency for AI inference at the edge.

Your Role and Responsibilities

This role is part of the EdgeCortix’s SAKURA AI Chip team

Work closely with our compiler and hardware engineers to accelerate the adoption of state-of-the-art neural networks on EdgeCortix's proprietary hardware and software stack. Engage in customer-oriented activities guiding deploying of custom and out-of-the-shelf models to meet accuracy and performance targets of customer’s AI system. Grow and maintain a collection of ready-to-use models for an ever-growing set of perception tasks. Work closely with our compiler team to continuously improve stability and user-experience of our software stack.

Your day-to-day activities will include but are not limited to dealing with existing neural network implementations, implementing models from scratch, training, quantization, pruning, deployment of models on EdgeCortix hardware, committing accuracy studies, reviewing academic literature, working with public and proprietary vision datasets, constantly communicating with hardware and compiler teams. You will use industry-standard PyTorch and TensorFlow frameworks and do a lot of Python programming.

Desired Qualifications:

• Experience in writing Python code
• Experience using TensorFlow and PyTorch ML frameworks
• Strong knowledge of Deep Neural Networks
• Strong understanding of network implementation, training and accuracy evaluation techniques
• Strong debugging and analysis skills, for root causing complex issues
• Experience with git, and work with github/gitlab development flows

Preferred Qualification:

• Pursuing a Master in Computer Science, Data Science or similar
• Strong object oriented design and development skills
• Knowledge of SOTA neural network architectures for popular perception tasks (such as object detection, semantic segmentation, monocular depth estimation), and popular vision datasets (COCO, KTTI)
• Experience in usage of deep learning compiler frameworks such as TVM, Glow or XLA
• Experience in deployment models and development of ML applications on an embedded device
• Experience with network quantization and pruning techniques
• Experience with customizing network implementations and hyper parameter tuning.
• Good knowledge of RNN/LSTMs, and Transformer architectures. Experience with point cloud processing, sequence (such as NLP), and graph processing networks (GNN)
• Knowledge of Neural Architecture Search (NAS) techniques is a plus
• Knowledge of Reinforcement Learning is a plus

What’s in it for you?

Make a difference: you will have the opportunity to join a well-funded fabless AI semiconductor startup that is disrupting the AI software and hardware co-design space. Be an integral part of its growth and momentum.

Location

Tokyo is the primary work location.

And Hybrid Remote working is allowed.

Benefits and Perks

• Highly competitive salary and stock options
• Flex work time
• Top-tier employee benefits