Remote Senior Deep Learning Engineer

Summary

Join Untether AI's top-notch team of engineers and contribute to the design, development, and verification of software interacting with our cutting-edge AI chip. Collaborate with hardware engineers and fellow software engineers to create software that maximizes the chip's capabilities, bringing AI inference to a wider audience. As a Senior Deep Learning Engineer, you will build tools for customers to optimize and deploy neural networks on our hardware, working with deep learning frameworks and contributing to SDK development. This role involves identifying and resolving performance bottlenecks to ensure scalable and reliable software. You'll be at the forefront of AI innovation, working on complex software and hardware designs. Untether AI offers a welcoming and diverse work environment.

Requirements

• Computer Science, Engineering or related degree
• 5+ years experience, with a healthy amount of both deep learning and software engineering (2+ years each)
• Experience with one or more deep learning frameworks (preferably PyTorch and/or ONNX)
• Experience working with one or more of the following: CNNs, LSTMs/RNNs, Transformers
• Experience reproducing academic work from the field of AI
• Strong math background
• Strong Python skills

Responsibilities

• Design, develop, and verify software that interacts with the AI chip
• Collaborate with hardware engineers and fellow software engineers
• Create software that fully realizes the capabilities of the hardware
• Identify and alleviate functional/performance bottlenecks to achieve scalable and reliable software
• Build tools for customers to ingest, optimize, and quantize their networks
• Work with deep learning frameworks at a deep level, modifying and extending backend functionality
• Support the development of the SDK to produce highly optimized, accurate networks running on the chip

Preferred Qualifications

• Masters or PhD in Computer Science, Engineering, or related field
• Embedded or low-level programming experience
• CUDA/OpenGL experience
• Experience deploying neural networks to production
• Experience with model compression techniques such as quantization, quantization-aware training, knowledge distillation, pruning, etc
• Experience with C/C++

Benefits

• 20 vacation days
• Strong health and extended health benefits
• Unlimited sick days
• Stock options