Jr AI Engineer

Rockstar

Posted 3 months ago

Full Time

,

Remote OK

Smart Summary

Responsibilities

The Jr. AI Engineer will support the development and testing of AI-powered features, focusing on prompt experimentation and RAG pipeline maintenance. They will also be responsible for data preparation, model evaluation, and writing clean Python code for internal tools and prototypes.

Qualifications

You have solid Python programming skills and a foundational understanding of machine learning, deep learning, NLP, and data processing concepts. You are comfortable using tools like PyTorch, TensorFlow, scikit-learn, and Hugging Face, and possess an interest in LLMs, GenAI systems, and prompt engineering.

Must Have Skills for ATS

Python

machine learning

deep learning

NLP

data processing

PyTorch

TensorFlow

scikit-learn

Hugging Face Transformers

pandas

NumPy

LLMs

GenAI

prompt engineering

RAG

embeddings

semantic search

Git

APIs

Docker

CI/CD

Job Description

Rockstar is recruiting for a data intelligence platform company focused on security analytics, investigations, fraud detection, and enterprise AI systems. Their team is dedicated to building production AI products that help organizations extract actionable insights from complex data. They are seeking a Jr. AI Engineer to contribute to their growing AI capabilities.

Position Summary

Our client is seeking a Jr. AI Engineer/Jr. Machine Learning Engineer to support the development, testing, and improvement of AI-powered features across their data intelligence platform. This role is designed for an early-career engineer who has strong technical fundamentals, curiosity about GenAI systems, and an interest in learning how production AI products are built and maintained.

The Jr. AI Engineer will work closely with senior engineers to assist with prompt experimentation, data preparation, RAG pipeline support, model evaluation, documentation, debugging, and basic AI service development. This role offers hands-on exposure to LLMs, embeddings, retrieval systems, ML workflows, and production engineering practices.

Essential Responsibilities
  • Assist in developing AI-powered features using Python, LLM tools, ML libraries, APIs, and internal platform services.
  • Support prompt engineering, prompt testing, model comparison, and evaluation of AI-generated outputs.
  • Help build and maintain RAG workflows, including document preparation, chunking, metadata tagging, embedding generation, retrieval testing, and result review.
  • Prepare, clean, format, and validate datasets used for model testing, prompt evaluation, and AI experiments.
  • Assist with model and workflow evaluation by reviewing outputs, identifying errors, documenting patterns, and comparing performance across approaches.
  • Write clean, readable Python code for scripts, internal tools, prototypes, experiments, and service components.
  • Support debugging of AI workflows, data pipelines, API integrations, and model behavior under the guidance of senior engineers.
  • Participate in code reviews, design discussions, team planning, and documentation efforts.
  • Learn and apply production engineering practices, including Git workflows, testing, logging, Docker, CI/CD, and deployment basics.
  • Document experiments, implementation details, findings, and recommendations clearly for technical team members.
Required Qualifications
  • 0–2 years of experience in AI engineering, machine learning, software engineering, data science, or a related technical area.
  • Internship experience, academic work, bootcamp projects, portfolio projects, or open-source contributions are acceptable.
  • Solid Python programming skills.
  • Foundational understanding of machine learning, deep learning, NLP, data processing, and model evaluation concepts.
  • Familiarity with tools or libraries such as PyTorch, TensorFlow, scikit-learn, Hugging Face Transformers, pandas, NumPy, or similar technologies.
  • Interest in LLMs, GenAI systems, prompt engineering, embeddings, semantic search, RAG, and AI agents.
  • Ability to work with structured and unstructured data.
  • Comfort using Git, notebooks, command-line tools, APIs, and collaborative development workflows.
  • Strong attention to detail, curiosity, problem-solving ability, and willingness to learn from feedback.
  • Clear written communication skills for documenting technical work and experiment results.
Preferred Qualifications
  • Portfolio, academic, internship, or project experience involving LLMs, chatbots, semantic search, classification, summarization, automation, or ML workflows.
  • Exposure to vector databases, embeddings, document processing, information retrieval, or search systems.
  • Familiarity with Docker, cloud environments, CI/CD concepts, or basic deployment workflows.
  • Exposure to agent frameworks such as LangGraph, AutoGen, CrewAI, or similar tools.
  • Coursework or practical experience in machine learning, NLP, statistics, data engineering, computer science, or software engineering.
  • Interest in security analytics, investigations, data intelligence, fraud detection, or enterprise AI systems.
Special Skills or Experience Required
  • Foundational knowledge of machine learning, deep learning, NLP, LLMs, prompt engineering, and RAG concepts.
  • Solid Python skills with exposure to ML libraries such as PyTorch, TensorFlow, scikit-learn, Hugging Face Transformers, or similar tools.
  • Experience through coursework, internships, projects, or portfolio work involving AI, data preparation, model testing, search, or automation.
  • Ability to document experiments, compare model outputs, support debugging, and learn production ML practices such as Git, APIs, Docker, and CI/CD.
Success Measures

Success in this role will be measured by consistent contribution to AI experiments, clean and reliable implementation work, clear documentation, improved evaluation support, effective debugging assistance, and steady growth in production AI engineering skills. The role should help increase team capacity while developing strong internal AI engineering talent over time.

Rockstar

Founded by entrepreneur Eddie Phanichkul in 2014, the Rockstar Art & Science Education Advancement Foundation is a future 501(c)(3) nonprofit corporation based in San Diego, CA. Our immediate mission is to use music to educate underprivileged children. We’ll do this by creating hands-on experiences and performances for children to get them excited about music, and then ultimately science and arts and beyond. Our secondary mission is to create grant and scholarship programs designed to help youth and young adults afford tools for science, technology, art, and music, while offering low-cost or no-cost workshops designed to reinforce and educate youth and young adults on conceptual science, technology, art, and music.
Runway Icon
Boost Your Interview Chances

With Runway

See Your Fit for This Role

1-5 min

Your Score

?

Top Applicants

90%

Your Job Search Advantage

Key Gaps & Next Steps:

Address these in your resume & Interview

Top Strengths For This Role

Highlight these in your cover letter & interview

Your Interview Guide

A Personalized Interview Strategy

Freshest Opportunities

Never Miss a Good Fit

Get notified when jobs mach your criteria