Junior ML Engineer — Generative AI (LangChain, LLM Swarms, Diffusion) in DataRoot Labs

We present an engaging opportunity to explore and work with various technologies beyond ChatGPT prompts. You’ll have the flexibility to transition between teams, enabling you to contribute to diverse projects, including complex LLM training pipelines, 3D generative AI, diffusion models, and multimodal edge ML.

Our team comprises experts in the foundational elements of next-generation AI systems, encompassing multimodal inference, generative reasoning, and robust training pipelines. We also maintain an in-house GPU cluster, providing resources for interacting, training, and experimenting with larger models for both professional projects and personal exploration.

In addition to working on client projects, you’ll have the chance to engage with our product development, including edge multimodal ML, GPT swarms, reinforcement learning, and much more, depending on your areas of interest.

Your initial project will involve building LLM agents, a less complex task that will offer a gentle introduction to our operations.

Joining our team — means joining a knowledgeable group of researchers and engineers in an environment eager to stay at the forefront of AI innovation, a critical endeavor in today’s rapidly advancing AI landscape.

Requirements:

• Strong Python Experience (1 year +)
• Specialized experience with LLMs, LangChain, RAG, and related technologies
• Knowledge of various machine learning methodologies, including classic machine learning, recommendation engines, natural language processing, computer vision, and beyond
• Experience in building REST APIs, knowledge of Docker, relational & vector DBs
• Familiarity with ML deployment techniques, emphasizing SD / LM checkpoints
• At least upper intermediate written/spoken English
• Be able to visit the office at least 3 times a week

Would be a plus:

• Knowledge of existing yet fresh AI tools, AI marketplaces, etc.
• C/C++ knowledge
• Experience in edge ML

Key Responsibilities:

• Tackle tough problems and find creative ways to solve them
• Maintain and optimize the solution for maximum speed and flexibility
• Guide, tune, and train big neural nets 🙂
• Write clean and reusable code
• Improve your skills and stay up-to-date with emerging technologies

What We Offer:

• Startup culture, a strong goal-oriented team, and a research mindset
• Opportunity to leverage your engineering skills for fellow engineers and shape the future of AI
• Working with the newest technical equipment (Latest MacBook Pro)
• 20 working days of annual vacation leave
• English courses, Educational Events & Conferences
• Medical insurance