Skip to content

Latest ML Engineering jobs at TensorStax

TensorStax

Research Engineer Intern, Evaluations

Designs evaluation frameworks and benchmarks to test AI agents' autonomy, reasoning, and reliability in data pipelines and warehouses. Requires experience in LLM benchmarking, reinforcement learning, Python, PyTorch/JAX, and data engineering tools.

San Francisco, CAML EngineeringHybridJAXdbt
TensorStax

Research Engineer, Reinforcement Learning

Develops RL environments and fine-tunes language models using PPO, DPO, and KTO to enhance agentic capabilities for data infrastructure tasks. Requires deep RL expertise, LLM fine-tuning knowledge, and strong problem-solving skills.

San Francisco, CAML EngineeringHybridPpoDpo