About the role#
TensorStax is building autonomous AI systems to manage and optimize mission-critical data infrastructure. We integrate reinforcement learning and language models to enhance reasoning over large-scale data lakes and warehouses. We are looking for a Research Engineer Intern to design evaluation frameworks and benchmarks that assess the autonomy, adaptability, and reliability of AI agents in data engineering environments.
What you'll do#
- Develop evaluation environments to test AI agents' ability to reason, plan, and act autonomously within mission-critical data pipelines.
- Design benchmarks to assess model capabilities in failure detection, pipeline optimization, and agentic decision-making in data workflows.
- Implement automated assessment frameworks for language model-based agents operating over data lakes and warehouses.
- Work with synthetic and real-world datasets to create testing environments for AI-driven data automation.
- Collaborate with research engineers to refine reward shaping strategies, guiding models toward efficient and agentic behaviors in data-intensive tasks.
What you'll need#
- Experience in language model research, specifically benchmarking in mission-critical domains.
- Strong background in AI evaluation methodologies, reinforcement learning, and RLHF techniques.
- Familiarity with benchmarking language models for structured and unstructured data tasks.
- Proficiency in Python and experience with ML frameworks like PyTorch or JAX.
- Hands-on experience with data lakes, warehouses, and data engineering tools.
- High agency and a proactive approach to research.
- Attention to detail when designing rigorous and reproducible experiments.
- Pursuing a degree in Computer Science, Computer Engineering, Electrical Engineering, Data Science, Mathematics, or Statistics.
Location & details#
- This position is based in San Francisco, California, and follows a hybrid work model.
- This is a full-time, paid internship with a rolling term.
- We do not offer visa sponsorship for this role.
About TensorStax
TensorStax is a software development company based in San Francisco. It builds autonomous AI solutions for data pipeline maintenance. The company works with existing infrastructure to optimize data workflows. It serves markets that require efficient data management. The team consists of 2 to 10 employees.
How to get in at TensorStax
Applying early to TensorStax gives you a distinct advantage because recruiters review applications before the pile grows. Intern Insider sends an instant alert the moment a role matching your target is published, helping you apply among the first. This timing often makes the difference in getting your resume seen. You can also improve your response rates by reaching out to the recruiters behind these roles directly. Intern Insider surfaces the specific people managing hiring at TensorStax so you can ask about the role or a referral. This direct approach is more effective than relying on a standard application queue.
