Open SourcePythonPyTorchTensorflowAIMachine LearningMLComputer VisionGenerative AILarge Language ModelsOpenAIAnthropicGeminiRAGTensorFlowMLOpsJAXRemote Work
About this role
Role Overview
Design and implement generative AI solutions for generating and editing images, text, and templates
Integrate and orchestrate large language models (LLMs) and vision models into design products
Develop scalable, low-latency inference pipelines for production use
Collaborate with product and engineering teams to define Trakto's AI roadmap
Evaluate, experiment with, and select the most suitable AI models and frameworks for each use case
Ensure quality, monitoring, and observability of models in production
Contribute to the evolution of the company’s data architecture and MLOps
Mentor and guide other engineers on AI development best practices.
Requirements
5+ years of software engineering experience, with at least 2 years focused on AI/ML in production
Strong proficiency in Python and machine learning frameworks (PyTorch, TensorFlow, or JAX)
Experience with LLMs, generative AI APIs (OpenAI, Gemini, Anthropic, Stability AI, etc.) and advanced prompting techniques
Knowledge of computer vision and image-generation models (Stable Diffusion, DALL·E, Imagen, etc.)
Experience with cloud AI infrastructure
Ability to translate business problems into viable, scalable AI solutions
Bonus if you:
Experience with RAG (Retrieval-Augmented Generation), fine-tuning, and model adaptation
Knowledge of MLOps: model versioning, drift monitoring, and retraining pipelines
Contributions to open-source AI projects or academic publications
Experience with creative products, design tools, or content generation at scale
Fluent or advanced English (to work with technical documentation and partners such as Google).