Senior Staff Deep Learning Engineer

Shield; City of Melbourne; On-site; AUD 110,000 - 160,000; Full time

Job summary

Shield AI in Melbourne seeks an experienced deep learning researcher to design and deploy perception systems, bridging research and field-ready solutions.

You will work on edge AI stacks, collaborate with DL engineers to scale DL operations, and apply pruning/quantization for embedded deployments while staying current with the literature.

Qualifications

• Demonstrable experience delivering deep-learning-based CV solutions (3–5 years).
• Strong understanding of CNNs and/or transformers for classification/recognition/segmentation.
• Experience with recent Foundation Models.

Responsibilities

• Research, design and implement state-of-the-art perception capabilities, taking ideas from conception into world-class field solutions.
• Work with and deploy our AI stack to edge devices
• Collaborate with other deep learning engineers to architect and develop tools to scale up deep learning operations
• Stay abreast with the literature and actively involve in various R&D project(s)

Skills

Deep learning

Computer vision

Edge deployment

Transformers

CNNs

Model pruning

Education

CS degree (BSc/MSc/PhD)

Tools

TensorFlow

PyTorch

Caffe

ONNX

Job description

Research, design and implement state-of-the-art perception capabilities, taking ideas from conception into world-class field solutions. Work with and deploy AI stack to edge devices, collaborate with other deep learning engineers to architect and develop tools to scale up deep learning operations, and stay abreast with the literature while actively involving in various R&D projects.

Key responsibilities

Research, design and implement state-of-the-art perception capabilities, taking ideas from conception into world-class field solutions

Work with and deploy our AI stack to edge devices

Work in collaboration with the other deep learning engineers to architect and develop tools to help scale up our deep learning operations

Stay abreast with the literature and actively involve in various R&D project(s)

About you

Demonstrable experience in delivering deep-learning-based solutions to solve computer vision problems with industry-based experience between 3-5 years

Strong understanding of using convolutional neural networks and/or transformers for object classification, recognition or segmentation

Experience working with recent Foundation Models

Experience with implementing novel deep learning network architectures using existing frameworks (TensorFlow, Caffe, PyTorch or similar)

Relevant tertiary qualifications (Bachelors/Master/PhD in Computer Science or related fields)

Publication(s) in world-leading Computer Vision/Artificial Intelligence/Machine Learning conferences/journals (i.e., CVPR, ICCV, ECCV, NeurIPS, ICLR, ICML, PAMI, JMLR) - preferred

In-depth understanding of the latest deep learning network architectures for computer vision and image processing - preferred

Experience deploying deep learning models in an embedded production context, including experience of structured and unstructured pruning, network quantization and performance tuning - preferred

About us

Shield AI is a venture-backed defense-tech company with the mission of protecting service members and civilians with intelligent systems. Its products include Hivemind autonomy software, V-BAT and X-BAT aircraft, and Aechelon simulation and synthetic reality technologies. With offices and facilities across the U.S., Europe, the Middle East, and Asia-Pacific, Shield AI's technology actively supports operations worldwide.