ML Solution Architect (Early Talent)

Responsibilities

• Help build and test LLM-based solutions and applications using Token Factory's inference services, including multimodal models (text, vision, audio).
• Aid senior SAs with prompt engineering, model selection, benchmarking, and inference optimization.
• Run performance and quality experiments to support proof-of-concept work.
• Contribute to internal tooling and automation that improves how the SA team delivers.

Requirements

• Currently pursuing or recently completed a BSc/MSc/PhD in Computer Science, Machine Learning, or a related field.
• Strong Python programming skills.
• Hands-on generative AI experience, including with common ML frameworks (e.g., PyTorch, Transformers).
• Strong communication skills, with a willingness to explain technical concepts to diverse audiences.

Qualifications

• Experience deploying/serving LLMs with vLLM, SGLang, or TensorRT-LLM (nice-to-have).
• Familiarity with inference optimization techniques such as quantization, batching, caching, and routing (nice-to-have).
• Knowledge of model architectures and fine-tuning approaches (nice-to-have).
• Contributions to open-source ML/AI projects (nice-to-have).

Skills

• Programming Languages – Python.
• ML Frameworks and Libraries – vLLM, SGLang, TensorRT-LLM, Transformers, OpenAI/Anthropic SDKs.
• Frameworks for Agentic Pipelines – Langchain / Langsmith / smolagents / equivalent.
• API and Web Frameworks – FastAPI, Flask.
• MLOps and DevOps tools – Kubernetes (K8s), Docker, Git.
• Cloud Platforms – AWS (SageMaker, Bedrock), GCP (Vertex AI), Azure (Azure ML).

Benefits

• Health insurance: 100% company-paid medical, dental, and vision coverage for employees and families.
• 401(k) plan: Up to 4% company match with immediate vesting.
• Parental leave: 20 weeks paid for primary caregivers, 12 weeks for secondary caregivers.
• Remote work reimbursement: Up to $85/month for mobile and internet.
• Disability & life insurance: Company-paid short-term, long-term and life insurance coverage.

Pay

Base Compensation Range $102—$126 USD

Schedule

Temporary contract for 3 months