Deep Learning Engineer

AI21 Labs; Brasil; Presencial; BRL 120.000 - 180.000; Tempo integral

Resumo da oferta

A leading AI technology company is seeking a Deep Learning Engineer to develop large language models and optimize training infrastructures. Ideal candidates will have significant experience in software development for production systems, deep learning, and proficiency with cloud platforms. Join us to advance your career in cutting-edge AI systems.

Qualificações

• 5+ years experience developing software for production systems or internal infrastructure/tools.
• Self learner with proven ability to remove technical road-blocks.
• Hands-on experience in deep learning and machine learning frameworks.

Responsabilidades

• Develop Large Language Models as part of applied research projects.
• Implement, optimize, scale, and test new ideas and architectures.
• Perform large-scale evaluations and comparisons of trained models.

Conhecimentos

Deep Learning

Production-grade Python

Cloud platforms (AWS, GCP)

TensorFlow

PyTorch

Formação académica

B.Sc. in computer science or software engineering

Ferramentas

Docker

Kubernetes

Descrição da oferta de emprego

Overview

• Engineering
• Tel Aviv, Israel
• Full-time

Description

Our team is looking for a Deep Learning Engineer.

AI21 is one of the few companies to have trained multi-billion parameter Large Language Models (LLMs), a feat that involves the most advanced engineering (large scale distributed training on thousands of cores). Serving these LLMs efficiently requires cutting-edge technology as well. As a deep learning engineer on the team, you will be responsible for maintaining and improving our training infrastructure, developing/scaling/testing new ideas, and adapting our code to run on and best utilize the newest and most advanced hardware accelerators.

Role and Responsibilities

• Develop Large Language Models as part of our applied research projects and in support of AI21 Platform, including designing, implementing and training massive-scale deep language models
• Implement, optimize, scale and test new cutting edge ideas and architectures
• Perform large-scale evaluations and comparisons of trained models across a range of benchmarks, as well as adding support for new benchmarks

Requirements

• B.Sc. in computer science, software engineering or equivalent
• Self learner, and proven record of ability to remove technical road-blocks
• 5+ years experience developing software for production systems and/or internal infrastructure/tools
• Prior experience working with cloud computing platforms (e.g. AWS, GCP, Docker, Kubernetes)
• Skilled at writing production-grade Python code
• Hands-on experience in deep learning and machine learning (TensorFlow/PyTorch..)
• Any one of the following:

• Optimization of deep learning model training (E.g. parallelization, megatron, deepspeed, FSDP)
• Custom kernel experience (C++/CUDA and/or Triton)
• Distributed Systems, in particular distributed deep learning training/serving

About Us

AI21 Labs is pioneering the development of Foundation Models and AI Systems for enterprises, accelerating the adoption of Generative AI in production.

Established in 2017 by AI visionaries Prof. Amnon Shashua, Prof. Yoav Shoham, and Ori Goshen, our mission is to equip businesses with cutting-edge LLMs and AI capabilities. Backed by leading investors like Pitango, Google, Nvidia, Intel Capital, and Comcast Ventures.

Join us on this exciting journey and advance your career with AI21 Labs!