AI Researcher

Trax Retail; Kraków; Hybrid; PLN 312,000 - 391,000; Full time

Benefits offered by this job

Remote or Hybrid work from Krakow

Medical package

English language courses

Work equipment provided

Career development

Learning resources

Job summary

Trax Retail is seeking an AI Software Engineer to develop production-grade ML/DL models and vision-language systems for global retail imagery. You will work on AR-enabled, edge-deployed solutions, balancing algorithmic innovation with robust, scalable software practices.

You will join an engineering-first team with strong emphasis on code quality, testing, and cross-functional collaboration in an international setting.

Qualifications

• Experience: 5+ years of hands-on experience developing and deploying ML/DL models in production or applied R&D.
• CV & math foundations: strong vision, algebra, calculus, 2D/3D geometry and graphics.
• Multimodal & Generative AI: practical experience with LLMs/VLMs and prompting techniques.
• Core tech: Python, Keras, TensorFlow, JAX, and standard software tools.
• Research-to-Product mindset: ability to turn ideas into robust production features.
• Communication: fluent English in a distributed international team.
• Software engineering practices: clean code, tests, Git, Jira, CI/CD.

Responsibilities

• Write clean, maintainable, and fully-tested production code with strong code reviews.
• Develop and optimize deep learning models and VLMs for retail scenes.
• Design AR and spatial features bridging CV and graphics for real-time use.
• Profile and optimize edge and on-device deployment for latency constraints.
• Experiment with multimodal LLMs and prompting techniques for automation.
• Improve algorithms for robustness and global scalability.

Skills

Python

JAX

TensorFlow

Keras

Computer Vision

Multimodal AI

Git

Jira

CI/CD

Edge Deployment

Tools

Jira

Git

CI/CD Pipelines

Job description

About the Role

Our R&D team builds cutting-edge visual intelligence for the global retail industry, analyzing complex in-store scenes and dense shelf environments across major markets worldwide. We combine deep learning, modern Vision-Language Models (VLMs), and generative AI to extract real-time understanding from retail imagery at massive scale.

Description

About the Role: Our R&D team builds cutting-edge visual intelligence for the global retail industry, analyzing complex in-store scenes and dense shelf environments across major markets worldwide. We combine deep learning, modern Vision-Language Models (VLMs), and generative AI to extract real-time understanding from retail imagery at massive scale.

Recently, we expanded into Augmented Reality (AR) - merging machine learning, 2D/3D geometry, and computer graphics to deliver interactive, spatial experiences directly onto edge devices. As an AI Software Engineer on this team, you will focus on solving non-trivial algorithmic challenges, developing visual and multimodal architectures, and shipping production-grade solutions deployed across thousands of stores globally. Crucially, this is an engineering-first role: roughly 80% of your time will be spent writing high-quality, fully-tested production code rather than conducting pure academic research.

About Form

FORM powers the world’s two billion mobile workers as they change companies and industries for good, with mobile technology that improves execution from the frontline. FORM solutions include the AI-enabled task management platform GoSpotCheck, Trax’s image recognition technology, and FORM OpX, all of which activate and connect teams in the field – with leaders, missions, and each other – so they can deliver success in the enterprise. With more than 25 years of experience, FORM supports 100,000+ global users from some of the world’s more recognizable global brands in 45 countries.

Requirements

Key Responsibilities

• Software Engineering & Code Quality: Write clean, maintainable, and fully-tested code. Actively participate in code reviews and collaborate closely with the team using Agile methodologies (Jira) and version control (Git).
• Model Development & Optimization: Train and evaluate deep learning models and Vision-Language Models (VLMs) tailored for fine-grained retail scene understanding and product recognition.
• AR & Spatial Solutions: Design and implement features bridging computer vision and computer graphics, applying 2D/3D geometric principles to power interactive Augmented Reality workflows.
• Edge & On-Device Deployment: Profile, optimize, and quantize non-trivial algorithms for latency- and memory-constrained edge hardware (e.g., mobile devices and embedded systems).
• VLM & Generative AI Innovation: Work hands-on with multimodal LLMs, VLMs, and advanced prompting techniques to create next-generation automated recognition tools.
• Algorithm Refinement & Scaling: Enhance and optimize existing algorithms to ensure high efficiency, robustness against varying store conditions, and global scalability through rigorous software architecture.

Requirements & Qualifications

• Experience: 5+ years of hands-on experience developing and deploying Machine Learning and Deep Learning models in production or applied R&D environments.
• Computer Vision & Math Foundations: Strong understanding of classical and modern computer vision, along with the linear algebra, calculus, and mathematical foundations required for 2D/3D geometry and spatial graphics.
• Multimodal & Generative AI: Practical experience working with modern LLMs/VLMs and prompt engineering techniques.
• Core Tech Stack: High proficiency in Python, deep learning frameworks (Keras, Tensorflow, Jax preferred), and standard software development tools.
• Research-to-Product Mindset: Strong problem-solving ability to take complex academic/R&D concepts and adapt them into robust, non-trivial, and highly maintainable production features.
• Communication: Comfortable collaborating daily in English, both written and verbal, in a distributed international team.
• Software Engineering Best Practices: Strong commitment to writing clean code, unit/integration tests, and using Git. Experience working within a structured team environment (code reviews, Jira, CI/CD pipelines).

Nice to Have

• Experience with Augmented Reality frameworks (e.g., ARKit, ARCore) or computer graphics pipelines (OpenGL, WebGL, or game engines).
• Edge Computing: Familiarity with optimizing models for edge devices and inference runtimes (e.g., CoreML, or TFLite).
• Prior domain experience in retail analytics, planogram compliance, or fine-grained object recognition.

What We Offer

• Full remote or hybrid work from our cozy office in Krakow.
• The option to enroll in a medical package on preferential terms.
• Subsidized English language courses.
• Company-provided work equipment.
• Opportunities for internal growth and career development.
• Access to internal learning resources.

Compensation

28,000 - 35,000 PLN per month. Exact compensation may vary depending on skills, experience, and location within Poland.