AI/ML Engineer

Career Overview

‍

TaskForge is building the infrastructure layer for AI-powered workflows, and we are looking for an AI/ML Engineer to help us push the boundaries of what is possible. You will work on model integration, inference optimization, and the AI features that differentiate TaskForge in the market.

‍

‍

About the role

‍

This is a deeply technical role at the intersection of machine learning engineering and product development. You will design and implement the systems that connect third-party AI models to our pipeline engine, build our internal model evaluation framework, and work closely with product to shape the AI features our customers rely on. You will also lead research efforts to identify and prototype new AI capabilities for the platform.

‍

‍

What you will do

‍

Build and maintain the model integration layer that connects TaskForge to LLMs and specialized AI models. Design inference pipelines optimized for latency, throughput, and cost. Develop evaluation frameworks to measure model accuracy and output quality over time. Collaborate with product and design to define the AI-native features in our roadmap. Stay current with AI research and identify opportunities to apply new techniques to TaskForge's platform.

‍

‍

What we are looking for

‍

• 3+ years of experience in ML engineering or applied AI research.
• Strong Python programming skills and familiarity with PyTorch or JAX.
• Experience deploying models to production and operating inference infrastructure.
• Solid understanding of LLMs, prompt engineering, RAG architectures, and fine-tuning techniques.
• Ability to communicate complex AI concepts clearly to non-technical stakeholders.

‍

Bonus if you can

‍

Bring experience with quantization, distillation, or other model optimization techniques. Have published research or contributed to open source ML projects. Have worked with multi-modal models or real-time inference at high scale.

‍

‍

What you will get

‍

• Competitive compensation with meaningful equity.
• Remote-first with flexible hours across all time zones.
• Compute budget for experiments and research.
• Access to frontier AI models and APIs.
• Annual conference and research budget.
• Comprehensive health and wellness benefits.

‍

Why join our team

‍

You will work on AI problems that are embedded directly in a production product used by real teams, not just experiments in a notebook. We move fast and give you the space to explore and prototype ideas. Our team values scientific rigor and pragmatic engineering in equal measure.