With over 50 products and 1.5 billion installs worldwide, AIBY is at the forefront of mobile-first product development with a focus on AI technology. Innovation and team spirit are the backbones of our company, and thanks to them, we are proud of titles like ChatOn, Arta, Forma, Plantum, Fonts Art, and Themify. At AIBY, you can enjoy the work you do while helping millions of people.
To further expand our horizons and achieve even greater results in a number of fields, we’re looking for a Lead Research Scientist, who will stay updated with the latest industry trends, tools, and techniques in Generative AI and apply them to enhance our AI capabilities.
What you will do:
- Optimize deep neural networks for fast inference (distillation, quantization, research of new architectures, etc.)
- Train and fine-tune multi-modal diffusion models using custom datasets
- Deploy generative models using various inference engines like TensorRT, AITemplates, and CoreML
- Research and experiment to improve existing models and propose new approaches
- Work with the Engineering Team as well as visualize and report research results to the Product Team
Tech stack:
- Inference Engines & deploy
- TensorRT + NVidia Triton Server + FastAPI
- AITemplate + Lightning (FastAPI) + RabbitMQ
- CoreML – will be a plus
- ML frameworks: PyTorch + JAX
- Huggingface: diffusers, transformers, tokenizers
ML/DL approaches:
- Distillation, quantization
- Diffusion Models
- Generative Adversarial Networks (adversarial training, perceptual loss)
- Super Resolution
- Traditional Computer Vision: object detection, classification, semantic segmentation (will be a plus)
What you should have:
- Python and PyTorch proficiency
- Hands-on experience with the development process in DL and ML (Docker, git, TensorBoard, DVC, Weights & Biases)
- Experience with a similar stack to ours
- Understanding of modern computer vision concepts such as diffusion models, GANs, and LLMs
- At least a B1 level of English
Will be a plus:
- Public activity on Kaggle, GitHub, Leetcode or other platforms
- Software development background
- At least a B1 level of Belarussian/Ukrainian/Russian
- A PhD in Computer Vision or a relevant technical field
Why work with us?
- A competitive salary with generous bonuses
- A variety of unique projects
- Independent decision-making and no meaningless paperwork
- A chance to directly influence the development of our products
- An opportunity to work from anywhere
- Sports reimbursement program
- Health Benefits (depending on the location)
- Reimbursement for professional courses and training
- 21 working days of paid vacation
- 5 sick days and other benefits
Location
AIBY is a team of strong international top talents.This role allows for either hybrid (combining in-office and remote work), fully remote or in-office work.
