AI Researcher

Trax is seeking an AI Researcher to join our R&D team, where we build cutting-edge visual intelligence for the global retail industry. You will work on advanced deep learning, Vision-Language Models, and Augmented Reality solutions to deliver real-time insights at a massive scale.

Key responsibilities

• Train, fine-tune, and evaluate deep learning models and Vision-Language Models tailored for fine-grained retail scene understanding.
• Design and implement features bridging computer vision and computer graphics, applying 2D/3D geometric principles to power interactive Augmented Reality workflows.
• Profile, optimize, and quantize algorithms for latency- and memory-constrained edge hardware.
• Work hands-on with multimodal LLMs and advanced prompting techniques to create next-generation automated recognition tools.
• Systematically benchmark and evaluate ML models against real-world retail datasets to ensure top-tier accuracy.

Requirements

• 3+ years of hands-on experience developing and deploying Machine Learning and Deep Learning models in production or applied R&D environments.
• Strong understanding of classical and modern computer vision, linear algebra, and the mathematical foundations required for 2D/3D geometry.
• Practical experience working with modern LLMs, VLMs, and prompt engineering techniques.
• High proficiency in Python and deep learning frameworks such as Keras, Tensorflow, or Jax.
• Strong problem-solving ability to adapt complex R&D concepts into robust, production-grade features.

What we offer

• Fully remote work environment within a distributed international team.
• Opportunities for internal growth and professional career development.
• Access to comprehensive internal learning resources and training.
• Company-provided work equipment to support your daily tasks.