A

ai&

Builds enterprise AI platform with governance

Member of Technical Staff - Post Training

AI Research +1Confirmed live in the last 24 hours
Full-timeNo salary listed

About the Job

Requirements

  • Hands-on experience running reinforcement learning workflows on language models, including implementing reward models, handling reward hacking, tuning Kullback–Leibler penalties, and applying RL theory.
  • Hands-on experience with data generation and evaluation for large language model post-training, including supervised fine-tuning, preference alignment, and reinforcement learning workflows on real models.
  • Strong proficiency in Python and PyTorch, with experience optimizing training pipelines.
  • Experience with DeepSpeed, Fully Sharded Data Parallel, vLLM, or similar frameworks for efficient model training and inference.
  • Experience designing and executing data generation, filtering, curation, and quality assessment processes at scale.
  • Ability to reason across data generation, training, alignment, and evaluation as a unified system.
  • Comfort working directly with enterprise customers, translating customer needs for internal technical teams, and owning technical delivery.
  • Familiarity with continual and lifelong learning in neural networks, including catastrophic forgetting and methods for keeping models current without degrading.

Responsibilities

  • Profile, optimize, and scale reinforcement learning training runs to reduce iteration time.
  • Integrate new optimization techniques from the research community.
  • Design and implement training environments that test model capabilities and turn proof-of-concept ideas into production-ready pipelines.
  • Build and maintain post-training infrastructure, including supervised fine-tuning, preference alignment, reward model pipelines, experiment tracking, and evaluation infrastructure, for internal model development and enterprise engagements.
  • Own enterprise customer post-training engagements by translating requirements into post-training specifications, running workflows, designing task-specific evaluations, and feeding learnings into core pipelines.
  • Design and build synthetic data pipelines for post-training and reinforcement learning at scale, and own generation, filtering, and quality assessment workflows.
  • Develop methodologies and infrastructure for continual model improvement without catastrophic forgetting, including training regimes and evaluation protocols for ongoing development as data and feedback accumulate.
  • Design task-specific evaluations beyond standard benchmarks, interpret results, identify regressions before production, and use findings to drive improvements.
  • Contribute to post-training and reinforcement learning research and share findings with the broader community.

About the Company

ai&

Builds enterprise AI platform with governance

AI&D builds an enterprise AI platform and offers end-to-end AI services to help organizations adopt, govern, and deploy AI across operations, covering strategy, data governance, use-case discovery, and production implementation. It aligns AI with business goals through maturity assessments, data strategy, governance, readiness checks, proofs of concept, and hackathons, and it develops frontier models in language, video, and multimodal domains with continual learning to improve security and performance while integrating with client systems. It differentiates itself by offering a unified global platform across multiple modalities, with a strong emphasis on continual learning, data governance, and hands-on enablement tailored to enterprise environments rather than only building models. Its goal is to help enterprises transform operations and create measurable business value by embedding AI into core processes with productive, scalable, and secure AI-enabled systems.

Company size
11-50
Company stage
Seed
Total funding
$57.8M
Headquarters
Yokohama, Japan
Founded
2026

Recently Posted Jobs

Explore more opportunities at ai&

See more jobs