Lead AI Engineer

Lead AI Engineer You might be our missing piece if you have: Deep expertise in Python and AI frameworks such as PyTorch (ideal), TensorFlow, Keras, or SciPy. Production-grade experience with FastAPI, Flask, or Django for serving AI models and building API-driven applications. Proven experience in architecting and scaling ML systems in production—everything from data pipelines and model deployment to monitoring and optimization. Experience with AI/ML architectures and AI-powered product architectures including: multi-agent architectures, multi-headed models, inference architectures. Experience with connecting data pipelines and serving inference processes. Strong hands-on knowledge of Docker and an understanding of containerized ML workflows (DevOps builds it, but you speak its language). Proficiency in SQLAlchemy, Alembic, and relational databases, along with a solid understanding of how to model, move, and optimize data for ML use cases. Expertise in data wrangling and preprocessing using tools like Pandas, NumPy, or SciPy. Experience integrating with and building on external AI APIs (e.g., OpenAI, Anthropic, Hugging Face, etc.). A solid understanding of software engineering principles, versioning (Git), and Python code style (PEP 8). A track record of leading AI/ML projects from ideation to deployment. Experience mentoring and guiding engineers, helping them grow technically and strategically. We would be thrilled if you have: Exposure to research or academic papers, and a knack for translating theory into production-ready code. Comfort with cloud infrastructure (AWS, Azure, GCP, or DigitalOcean) and understanding of CI/CD best practices. Hands-on experience with LLMs, embeddings, semantic search, and tools like StableDiffusion or LLaMA. A collaborative mindset and the ability to bridge the gap between product, engineering, and research teams. We will be working together on the following: Understanding clients' and their requirements and providing optimal solutions. Staying ahead of industry trends and implementing cutting-edge solutions. Technical leadership over major AI initiatives, from architecture design to final delivery. Mentoring and upskilling the AI team—pair programming, code reviews, and continuous learning are your bread and butter. Driving best practices around model development, reproducibility, monitoring, and scalability. Staying up to date with state-of-the-art advancements and continuously leveling up our AI capabilities. Department AI & Data Role AI Engineer Locations Cluj-Napoca , Oradea , Brasov Remote status Hybrid Cluj-Napoca Oradea Brasov About RebelDot At RebelDot we enable organizations in more than 15 industries to make an asset out of custom software. From consulting to web or mobile apps, UX-UI design and QA, we help our clients achieve more through technology. Our goal is to make software development effective and hassle-free for small and medium enterprises. Helping our clients get the most value for their investment in technology is what drives us. Increasingly, this means working with them as a full technical partner, starting with an initial consulting stage where we understand their needs and propose the optimal approach – or, “the line”, as we call it. Because of our ‘rebel’ approach to software development, oftentimes, our solutions are very different from our peers as we stand out through innovation. From there on out, we partner up and lead the line for our clients. Find out more Founded in July, 2018 Co-workers 300+