Machine Learning Engineer

Hello, we’re Figr.

We’re on a mission to make the product process simpler and faster. We envision a world where creating meaningful products is accessible to everyone, not limited by technical skills or barriers.

About Us

Figr is a fast-growing, venture-funded startup developing design tools to automate workflows for product design teams. Figr aims to make the design process simpler—from research to ideation to prototyping.

We’re building foundational models that deeply understand interface design, generating high-quality first drafts of wireframes, high-fidelity interface designs, and user flows.

We’re not just building a prompt-to-design tool, but something that truly understands designers' needs, helping them create better products. With Figr, designers can focus on the creative work they love—while we handle the non-creative, mundane processes.

The Role

We are currently looking for an experienced Large Language Model (LLM) Engineer focused on fine-tuning and exploring LLMs for design purposes, including code generation for UI/UX. The role involves an end-to-end journey of creating data and fine-tuning open-source LLMs.

If you have a knack for problem-solving, possess excellent communication skills, and share our vision of redefining the democratization of AI, this role is for you!

Key Responsibilities

• Quality data curation and improvement for fine-tuning LLMs. This is the key responsibility, and we are looking for an out-of-the-box thinker who can brainstorm new ideas on how to tackle data-related problems.
• Fine-tuning LLMs.
• Deploying LLMs on GPUs with minimal latency on AWS/GCP.
• Leading the current Data Science team to the next breakthrough.

Skills Needed

• Excellent communication skills, allowing seamless collaboration within our diverse team.
• A minimum of 3-5 years of hands-on experience with PyTorch.
• Deep understanding of the Transformers architecture, its underlying components, and new SLM/LLM architectures.
• Proficiency in determining which metrics/loss functions to track for your use case.
• Experience with working on Lora, QLora, LLMs optimization techniques, and deployment strategies.
• Hands-on experience with multi-GPU training.
• Ability to implement research papers and stay updated with the latest trends that could be useful for our use case.
• Experience with Git and conducting code reviews.

Bonus Points For

• A dislike for the status quo and authority bias.

Apply

Fill out this form

and email at chirag@figr.design