AI Software Engineer

Trax is seeking an AI Software Engineer to join our R&D team, where we build cutting-edge visual intelligence for the global retail industry. You will work on complex in-store scenes, leveraging deep learning, Vision-Language Models, and generative AI to deliver production-grade solutions that impact thousands of stores worldwide.

Key responsibilities

• Develop and optimize 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 for Augmented Reality.
• Write clean, maintainable, and fully-tested production code while participating in code reviews and Agile workflows.
• Profile and quantize algorithms for deployment on latency-constrained edge hardware and mobile devices.
• Collaborate on the innovation of multimodal LLMs and advanced prompting techniques to create automated recognition tools.

Requirements

• 5+ years of hands-on experience developing and deploying Machine Learning and Deep Learning models in production environments.
• Strong mathematical foundations in linear algebra and calculus, specifically for 2D/3D geometry and spatial graphics.
• High proficiency in Python and deep learning frameworks such as Keras, Tensorflow, or Jax.
• Proven ability to adapt complex R&D concepts into robust, maintainable, and scalable production features.
• Excellent communication skills in English and a commitment to software engineering best practices like unit testing and Git.

What we offer

• Fully remote work environment within a distributed international team.
• Access to internal learning resources and opportunities for professional career development.
• Company-provided work equipment to support your daily tasks.
• Subsidized English language courses and optional medical package enrollment.