This list contains only the countries for which job offers have been published in the selected language (e.g., in the French version, only job offers written in French are displayed, and in the English version, only those in English).
As an AI Software Engineer you will be designing and optimizing our deep learning training framework and software tools for the development of data driven models. Your work will directly empower the AI Model Engineers to efficiently build and improve their solutions.
Job Responsibility
Designing and optimizing the training / inference pipelines and our deep learning framework
Developing robust Python code for preprocessing and training execution
Integrating software modules into our continuous integration and training pipeline
Profiling the performance of the training infrastructure, identifying and removing bottlenecks and improving scalability
Coordinating with the infrastructure and cloud teams to sustain high availability of the training environment
Working closely with our AI Hardware Engineers to ensure optimal compatibility of our software ecosystem to different chip architectures
Tracking the progress in open-source communities and other organizations on AI software frameworks to continuously improve our software landscape
Requirements
Advanced programming capabilities in Python
Understanding of software engineering concepts, OOD and design patterns
A good understanding of modern machine learning concepts and state of the art deep learning techniques such as CNN based architectures (e.g. YOLO, ResNet) or transformers (e.g. ViT, DeTR)
Experience in working with and scaling training frameworks, such as Tensorflow or PyTorch, preferentially in a cloud context
Experience with Python ML libraries such as numpy, pandas, scikit
Experience with training on 2D image data (e.g. camera)
Preferably: Experience with training on 3D pointcloud data (e.g LiDAR)
Preferably: Experience with embedded AI optimization
Preferably: Experience with visualizing radar data and experience with databricks
Nice to have
Experience with training on 3D pointcloud data (e.g LiDAR)
Experience with embedded AI optimization
Experience with visualizing radar data and experience with databricks
What we offer
Individual plan of learning and development, attending conferences, participation in company innovation program, language courses
Flexible working hours
Cafeteria benefit system (from 4 000 CZK/year)
Special reduced price of operating leasing for Škoda cars
Meal vouchers
Multisport card
Company contribution for supplementary pension insurance from 900 CZK/month
Sick days
Travel insurance for private purposes even for family members
T-Mobile Program – tariffs even for family members