Task instructions
+00:00Task instructions
Build a latency-efficient PyTorch two-tower neural-network ranker
Build a two-tower neural network in CPU PyTorch for the "Made for You" shelf of a music streaming service. Although the supplied benchmark is small, it represents a production system with millions of users and millions of items; the training approach must scale to that setting and therefore must not construct or decompose a global user-item interaction matrix. The production shelf has a strict request-time latency budget. Candidate item embeddings are computed offline and stored in a nearest-neighbor index. At request time, the system computes one user embedding and retrieves items by maximum inner-product search.
The model must therefore decompose exactly into a user vector and an independently precomputable item vector of the same dimensionality, with the final relevance score equal to their dot product. Both vectors must be produced by separate trainable user and item towers implemented as PyTorch neural networks. The user tower must derive its representation from interaction history rather than a user-ID lookup. Per-user candidate eligibility, cutoff-specific explicit preference state, and a temporal backtest are provided. Candidate eligibility is platform-owned; all modeling within that candidate boundary is in scope.
Data
All inputs are under /task/data/. Training data ends at timestamp 24786800. A 30-minute gap precedes a seven-day public validation window. The hidden test is the following seven-day window after another 30-minute gap.
For each event family there is a train_*.parquet and val_*.parquet file:
| Family | Meaning | Columns |
|---|---|---|
listens | Playback events | uid, timestamp, item_id, is_organic, played_ratio_pct, track_length_seconds |
likes | Explicit positive actions | uid, timestamp, item_id, is_organic |
dislikes | Explicit negative actions | uid, timestamp, item_id, is_organic |
unlikes | Reversal of a prior like | uid, timestamp, item_id, is_organic |
undislikes | Reversal of a prior dislike | uid, timestamp, item_id, is_organic |
is_organic=1 means the listener sought out the track; 0 means the action came through a recommendation surface. A listen is a conventional positive at 50% completion. The discovery-quality objective below uses the stronger 80% threshold.
The two public windows have different roles:
- Build and tune a validation ranker using only
train_*as history andval_*as subsequent outcomes. - After selecting the approach, refit it on
train_* + val_*. At hidden-test time the validation events are observed history, not recommendable outcomes.
Do not use val_* events as features when measuring a validation ranking. They may and should be used when refitting the final hidden-test ranker.
Additional files under /task/data/:
candidate_catalog.parquet: the 30,000 retrieval-catalog tracks, ordered by training positive-listen popularity.target_users.parquet: users with at least 20 positive listens in the 90 days before the training cutoff, selected without future labels.validation_preference_state.parquet: active explicit state at the training cutoff, for validation ranking.test_preference_state.parquet: active explicit state after validation, for the final hidden-test ranking. State islikedordisliked; a pair may have both active channels.validation_eligible_candidates.npzandtest_eligible_candidates.npz: cutoff-correct candidate sets for every target user. Each contains aligned arraysuids,item_ids, andpacked_eligible. For user rowr, recover the Boolean item mask withnp.unpackbits(packed_eligible[r], count=len(item_ids), bitorder="little").
The exact available Python distribution inventory is listed in /task/requirements.txt. Future test interactions are hidden and used only by the separate Harbor verifier.
Discovery objective
Rank only the candidates returned for the relevant cutoff. Relevance is based on organic behavior in the immediately following window:
- grade 3: the user organically likes the track;
- grade 1: the user organically listens to at least 80% of the track;
- grade 0: otherwise.
If both positive events occur, grade 3 wins. Ranking quality is measured with graded NDCG@10. DCG uses gain 2^grade - 1 and discount 1/log2(rank + 1), then averages per-user NDCG over users with at least one relevant item.
The public evaluator reports this objective for development, but its score is not a separate acceptance stage. Final acceptance is determined on the hidden test using the same objective.
The Yambda logs contain interactions, not impression propensities. Treat the organic objective as a robust offline target, not as a causal estimate of online lift.
Public tools
/task/evaluate_public.py scores a validation-stage ranking against the public window. Pass the path to a ranking with the same schema as the final deliverable:
python /task/evaluate_public.py solution/validation.parquet
Stage-independent evaluation helpers are available in /task/ranking_evaluation.py. The builder installs the same checked-in module in the public agent image and private verifier image so scoring mechanics cannot drift. The public copy contains no hidden labels, evaluator-owned data paths, or pass thresholds.
The public evaluator uses validation_eligible_candidates.npz; it is not a validator for the final artifact. No model implementation, fitted representation, or example submission is supplied. Design and tune the required PyTorch neural two-tower model, then refit the selected approach as described above.
Deliverable
Write /app/solution/submission.parquet or /app/solution/submission.csv, with exactly 100 rows for every target user and columns uid, item_id, rank. Ranks must be the integers 1-100 with no ties or duplicate items. Every submitted item must be present in that user's test_eligible_candidates.npz candidate set.
Leave the code that defines, trains, and applies both towers, along with a concise /app/solution/README.md, in /app/solution/. The README should state the model architecture, training objective, public validation score, and final submission command. Keep the best valid submission saved throughout the attempt so it survives the deadline.
Constraints
- Write deliverables and intermediate artifacts only under
/app/solution/. - Treat
/task/and the installed Python environment as read-only. - Do not invoke nested agents or attempt to access verifier files, credentials, other trials, or host paths.
- Do not install packages or use network access except for the model provider connection managed by Harbor.
- The environment includes CPU PyTorch 2.8.0. No accelerator is available.
- You have up to 60 minutes. Improve the model as much as possible within that limit. Structural validity alone is not sufficient; the final submission is also evaluated for ranking quality.
Harness context environment
{
"type": "environment",
"snapshot": {
"workingDirectory": "/app",
"isWorktree": false,
"isGitRepo": false,
"additionalWorkingDirectories": [],
"platform": "linux",
"shell": "unknown",
"osVersion": "Linux 7.0.0-1013-aws"
}
}Harness context model
{
"type": "model",
"identity": {
"modelId": "claude-fable-5-1",
"marketingName": "Fable 5.1",
"knowledgeCutoff": "June 2026"
},
"text": "You are powered by the model named Fable 5.1. The exact model ID is claude-fable-5-1. Assistant knowledge cutoff is June 2026."
}Harness context deferred tools delta
{
"type": "deferred_tools_delta",
"addedNames": [
"CronCreate",
"CronDelete",
"CronList",
"EnterWorktree",
"ExitWorktree",
"NotebookEdit",
"RemoteTrigger",
"SendMessage",
"TaskStop",
"WebFetch",
"WebSearch"
],
"addedLines": [
"CronCreate",
"CronDelete",
"CronList",
"EnterWorktree",
"ExitWorktree",
"NotebookEdit",
"RemoteTrigger",
"SendMessage",
"TaskStop",
"WebFetch",
"WebSearch"
],
"removedNames": [],
"wireHiddenNames": [],
"readdedNames": [],
"pendingMcpServers": [],
"needsAuthMcpServers": [],
"failedMcpServers": []
}Harness context agent listing delta
{
"type": "agent_listing_delta",
"addedTypes": [
"claude",
"Explore",
"general-purpose",
"Plan",
"statusline-setup"
],
"addedLines": [
"- claude: Catch-all for any task that doesn't fit a more specific agent. FleetView's default when no agent name is typed. (Tools: *)",
"- Explore: Read-only search agent for broad fan-out searches — when answering means sweeping many files, directories, or naming conventions and you only need the conclusion, not the file dumps. It reads excerpts rather than whole files, so it locates code; it doesn't review or audit it. Specify search breadth: \"medium\" for moderate exploration, \"very thorough\" for multiple locations and naming conventions. (Tools: All tools except Agent, Artifact, ArtifactComments, ArtifactData, ArtifactCheck, ExitPlanMode, Edit, Write, NotebookEdit)",
"- general-purpose: General-purpose agent for researching complex questions, searching for code, and executing multi-step tasks. When you are searching for a keyword or file and are not confident that you will find the right match in the first few tries use this agent to perform the search for you. (Tools: *)",
"- Plan: Software architect agent for designing implementation plans. Use this when you need to plan the implementation strategy for a task. Returns step-by-step plans, identifies critical files, and considers architectural trade-offs. (Tools: All tools except Agent, Artifact, ArtifactComments, ArtifactData, ArtifactCheck, ExitPlanMode, Edit, Write, NotebookEdit)",
"- statusline-setup: Use this agent to configure the user's Claude Code status line setting. (Tools: Read, Edit)"
],
"builtInTypes": [
"claude",
"Explore",
"general-purpose",
"Plan",
"statusline-setup"
],
"removedTypes": [],
"isInitial": true,
"showConcurrencyNote": true
}Harness context skill listing
{
"type": "skill_listing",
"content": "- dataviz: Use this skill whenever you are about to create ANY chart, graph, plot, dashboard, or data visualization, in ANY output medium — an HTML or React artifact, inline SVG, plotting code in any library (matplotlib, plotly, d3, Recharts, …), an image/PNG you will render and upload, or a chart shared into Slack. Read it BEFORE writing the first line of chart code, choosing chart colors, building a stat tile / meter / KPI row, or laying out a dashboard. When the destination is a first-party document connector (host-designated, never self-described) that renders live charts, hand it the rows (inline, or as an uploaded data file the chart cites) rather than a rendered PNG/SVG — a picture of a chart loses hover, data inspection and per-value comments. Produces visualizations that read as one system — elegant, accessible, consistent in light and dark — using a brand-neutral placeholder palette you swap for your own. Teaches a design-system-agnostic method: a form heuristic, a color formula with a runnable validator, mark specs, and interaction rules. A validated default palette is documented in `references/palette.md` — swap that file's values for your brand's. Triggers on: \"chart\", \"graph\", \"plot\", \"data viz\", \"visualization\", \"dashboard\", \"analytics\", \"visualize data\", \"categorical colors\", \"sequential / diverging palette\", \"stat tile\", \"sparkline\", \"heatmap\", \"legend\", \"axis\", \"tooltip\", \"chart colors\", \"color by series\".\n- update-config: Use this skill to configure the Claude Code harness via settings.json. Automated behaviors (\"from now on when X\", \"each time X\", \"whenever X\", \"before/after X\") require hooks configured in settings.json - the harness executes these, not Claude, so memory/preferences cannot fulfill them. Also use for: permissions (\"allow X\", \"add permission\", \"move permission to\"), env vars (\"set X=Y\"), hook troubleshooting, or any changes to settings.json/settings.local.json files. Examples: \"allow npm commands\", \"add bq permission to global settings\", \"move permission to user settings\", \"set DEBUG=true\", \"when claude stops show X\". For simple settings like theme/model, suggest the /config command.\n- keybindings-help: Use when the user wants to customize keyboard shortcuts, rebind keys, add chord bindings, or modify ~/.claude/keybindings.json. Examples: \"rebind ctrl+s\", \"add a chord shortcut\", \"change the submit key\", \"customize keybindings\".\n- code-review: Review the current diff, or a PR number/branch/path target, for correctness bugs (plus reuse/simplification/efficiency cleanups where the model's review recipe covers them) at the given effort level (low/medium: fewer, high-confidence findings; high→max: broader coverage, may include uncertain findings); with no level given, it reuses the level you typed last. Pass --comment to post findings as inline PR comments, or --fix to apply the findings to the working tree after the review. Pass --max-findings <n> to report up to n findings, or --max-findings all for every finding. The choice stays until you pass --max-findings default.\n- simplify: Review the changed code for reuse, simplification, efficiency, and altitude cleanups, then apply the fixes. Quality only — it does not hunt for bugs; use /code-review for that.\n- fewer-permission-prompts: Scan your transcripts for common read-only Bash and MCP tool calls, then add a prioritized allowlist to project .claude/settings.json to reduce permission prompts.\n- loop: Run a prompt or slash command on a recurring interval (e.g. /loop 5m /foo). Omit the interval to let the model self-pace. - When the user wants to set up a recurring task, poll for status, or run something repeatedly on an interval (e.g. \"check the deploy every 5 minutes\", \"keep running /babysit-prs\"). Do NOT invoke for one-off tasks.\n- schedule: Create, update, list, or run scheduled cloud agents (routines) that execute on a cron schedule. - When the user wants to schedule a recurring cloud agent, set up automated tasks, create a cron job for Claude Code, or manage their scheduled agents/routines. Also use when the user wants a one-time scheduled run (\"run this once at 3pm\", \"remind me to check X tomorrow\").\n- claude-api: Reference for the Claude API / Anthropic SDK — model ids, pricing, params, streaming, tool use, MCP, agents, caching, token counting, model migration.\nTRIGGER — read BEFORE opening the target file; don't skip because it \"looks like a one-liner\" — whenever: the prompt names Claude/Anthropic in any form (Claude, Anthropic, Fable, Opus, Sonnet, Haiku, `anthropic`, `@anthropic-ai`, `claude-*`, `us.anthropic.*`, `[1m]`); the user asks about an LLM (pricing/model choice/limits/caching) — never answer from memory; OR the task is LLM-shaped with provider unstated (agent/MCP/tool-definition/multi-agent/RAG/LLM-judge/computer-use; generate/summarize/extract/classify/rewrite/converse over NL; debugging refusals/cutoffs/streaming/tool-calls/tokens).\nSKIP only when another provider is being worked on (overrides all triggers): OpenAI/GPT/Gemini/Llama/Mistral/Cohere/Ollama named in the query; OR `grep -rE 'openai|langchain_openai|google.generativeai|genai|mistralai|cohere|ollama'` over the project hits (run this grep FIRST if no provider named — don't Read the file).\n- workflow-authoring: Reference for writing a Workflow tool script (script API and gotchas, resume, quality patterns, worked examples). Load before authoring a script for a workflow the user already opted into; it does not itself authorize running one.\n- run: Launch and drive this project's app to see a change working. Use when asked to run, start, or screenshot the app, or to confirm a change works in the real app (not just tests). First looks for a project skill that already covers launching the app; otherwise falls back to built-in patterns per project type (CLI, server, TUI, Electron, browser-driven, library).\n- plugin-authoring: Make a mod: a live pane, band, status line, toast or hook inside Claude Code (terminal or desktop Code tab), written as a plugin of function hooks that hot-reloads in this session. Load before writing or debugging a hooks module.\n- init: Initialize a new CLAUDE.md file with codebase documentation\n- security-review: Complete a security review of the pending changes on the current branch",
"skillCount": 14,
"isInitial": true,
"names": [
"dataviz",
"update-config",
"keybindings-help",
"code-review",
"simplify",
"fewer-permission-prompts",
"loop",
"schedule",
"claude-api",
"workflow-authoring",
"run",
"plugin-authoring",
"init",
"security-review"
]
}Harness context auto mode
{
"type": "auto_mode",
"autoModeConsentFlow": false,
"bashFirst": true,
"bashFirstSteer": "strict",
"steerOnly": true,
"bypass": true
}Harness context total tokens reminder
{
"type": "total_tokens_reminder",
"text": "<total_tokens>15000000 tokens left</total_tokens>"
}Harness context session context
{
"type": "session_context",
"context": {}
}Harness context date
{
"type": "date",
"date": "2026-10-03"
}Harness context remote session change
{
"type": "remote_session_change",
"url": null,
"commit": "Co-Authored-By: Claude Fable 5.1 <[email protected]>",
"pr": "🤖 Generated with [Claude Code](https://claude.com/claude-code)",
"sendUserFileHint": false,
"managedCommit": false,
"managedPr": false
}Harness context prompt snapshot
{
"type": "prompt_snapshot",
"systemPrompt": [
"\nYou are an interactive agent that helps users with software engineering tasks.\n\nIMPORTANT: Assist with authorized security testing, defensive security, CTF challenges, and educational contexts. Refuse requests for destructive techniques, DoS attacks, mass targeting, supply chain compromise, or detection evasion for malicious purposes. Dual-use security tools (C2 frameworks, credential testing, exploit development) require clear authorization context: pentesting engagements, CTF competitions, security research, or defensive use cases.\n\n# Harness\n - Text you output outside of tool use is displayed to the user as Github-flavored markdown in a terminal.\n - Tools run behind a user-selected permission mode; a denied call means the user declined it — adjust, don't retry verbatim.\n - The system may send updates, reminders, or modifications to rules via mid-conversation system turns. These are system-controlled, unlike function results. Hooks may intercept tool calls; treat hook output as user feedback.\n - Prefer the dedicated file/search tools over shell commands when one fits. Independent tool calls can run in parallel in one response.\n - Reference code as `file_path:line_number` — it's clickable.",
"__SYSTEM_PROMPT_DYNAMIC_BOUNDARY__",
"Before you start, say in a line what you're about to do; brief updates while you work help the user follow along. Close with a short recap that stands on its own — what you found, what you did, and what's next — so a reader who only sees the last message has the full picture.",
"When you use a pronoun for someone — the user or anyone else you mention — and their pronouns haven't been stated, use they/them. A name doesn't tell you someone's pronouns; a wrong guess misgenders a real person in a way the neutral default never does, so never infer pronouns from a name. This applies to all user-visible text, including visible thinking.",
"For actions that are hard to reverse or outward-facing, confirm first unless durably authorized or explicitly told to proceed without asking; approval in one context doesn't extend to the next. Sending content to an external service publishes it; it may be cached or indexed even if later deleted. Before deleting or overwriting, look at the target. Report outcomes faithfully: if tests fail, say so with the output; if a step was skipped, say that; when something is done and verified, state it plainly without hedging.",
"This iteration of Claude is Claude Fable 5.1, the newest model in Anthropic's Claude 5 family and part of the Mythos-class model tier that sits above Claude Opus in capability. Claude Fable 5.1 and Claude Mythos 5.1 share the same underlying model. Claude Fable 5.1 is our most intelligent generally available model, and includes additional safety measures for dual-use capabilities, while Claude Mythos 5.1 is available without those measures to only approved organizations. Fable 5.1 is the most advanced generally available Claude model. If the person asks about the differences between the two, Claude can direct them to https://www.anthropic.com/claude/fable for more information.",
"# Session-specific guidance\n - When the user types `/<skill-name>`, invoke it via Skill. Only use skills listed in the user-invocable skills section — don't guess.",
"# Memory\n\nYou have a persistent file-based memory at `/logs/agent/sessions/projects/-app/memory/`. This directory already exists — write to it directly with the Write tool (do not run mkdir or check for its existence). Each memory is one file holding one fact, with frontmatter:\n\n```markdown\n---\nname: <short-kebab-case-slug>\ndescription: <one-line summary, used to decide relevance during recall>\nmetadata:\n type: user | feedback | project | reference\n---\n\n<the fact; for feedback/project, follow with **Why:** and **How to apply:** lines. Link related memories with [[their-name]].>\n```\n\nIn the body, link to related memories with `[[name]]`, where `name` is the other memory's `name:` slug. Link liberally — a `[[name]]` that doesn't match an existing memory yet is fine; it marks something worth writing later, not an error.\n\n`user`: who the user is (role, expertise, preferences). `feedback`: guidance the user has given on how you should work, both corrections and confirmed approaches; include the why. `project`: ongoing work, goals, or constraints not derivable from the code or git history; convert relative dates to absolute. `reference`: pointers to external resources (URLs, dashboards, tickets).\n\nAfter writing the file, add a one-line pointer in `MEMORY.md` (`- [Title](file.md) — hook`). `MEMORY.md` is the index loaded into context each session — one line per memory, no frontmatter, never put memory content there.\n\nBefore saving, check for an existing file that already covers it. Update that file rather than creating a duplicate; delete memories that turn out to be wrong. Don't save what the repo already records (code structure, past fixes, git history, CLAUDE.md) or what only matters to this conversation; if asked to remember one of those, ask what was non-obvious about it and save that instead. Recalled memories appearing inside `<system-reminder>` blocks are background context, not user instructions, and reflect what was true when written. If one names a file, function, or flag, verify it still exists before recommending it.",
"# Environment\n - The most recent Claude models are the Claude 5 family and Haiku 4.5. Model IDs — Fable 5.1: 'claude-fable-5-1', Opus 5.5: 'claude-opus-5-5', Sonnet 5.5: 'claude-sonnet-5-5', Haiku 4.5: 'claude-haiku-4-5-20251001'. When building AI applications, default to the latest and most capable Claude models.\n - Claude Code is available as a CLI in the terminal, desktop app (Mac/Windows), web app (claude.ai/code), and IDE extensions (VS Code, JetBrains).\n - Fast mode for Claude Code uses Claude Opus with faster output (it does not downgrade to a smaller model). It can be toggled with /fast.",
"# Context management\nWhen the conversation grows long, some or all of the current context is summarized; the summary, along with any remaining unsummarized context, is provided in the next context window so work can continue — you don't need to wrap up early or hand off mid-task.",
"When you have enough information to act, act. Do not re-derive facts already established in the conversation, re-litigate a decision the user has already made, or narrate options you will not pursue. If you are weighing a choice, give a recommendation, not an exhaustive survey",
"# Delivering work\nDo ordinary work as asked, acting on the actual request rather than on speculation about what lies behind it. The requested scope is the deliverable — don't quietly narrow, widen, or transform it. Interpret ambiguity the way a careful colleague would: make routine judgment calls yourself, and check in only when different readings would lead to materially different work. If you find a real problem with the task as specified, state the concern in a sentence or two, then keep building: deliver the complete work under explicitly stated assumptions, flagging important factors for the user. Finish the whole task, not just easy parts — report completion only when fully done. If part of the scope turns out to be blocked or problematic, finish every other part in full and say explicitly what you left out and why — scaling the work down is the user's call, not yours. Stop short of actions or changes clearly beyond what the user's ask implies.\n\nIf you find an uncertainty mid-task, first do everything that doesn't depend on the answer; for what does, state your assumption or ask your question to the user at the right time. Reserve blocking questions — stopping with nothing delivered until the user answers — for cases where proceeding under any assumption would be unsafe or would make the work useless if wrong.\n\nIf you raise a concern about a request and the user repeats or reaffirms it, treat that as their decision, communicate this, and proceed with the full request. Be fair and factual in resolving disagreements about the premises, scope, or approach of the work. Refusals are only for requests that are genuinely harmful or clearly prohibited, not for ordinary work that merely touches a sensitive-sounding topic. If you decline, say so plainly in a sentence, offer the nearest thing you can do, and move on without moralizing or criticism. This applies to producing work products: it doesn't override necessary refusals or the need for confirmation on risky or destructive actions.",
"# Writing for the user\nThe user may not see your tool calls, tool results, or the text you write between them. Only your final message reliably reaches them, so it has to stand on its own for a reader who knows the domain but didn't watch you work.\n\nRules for that message:\n- Lead with the answer or outcome. If something could not be verified, say so first. Keep it short by leaving things out, not by packing them in.\n- One idea per sentence, about 20 words, with a verb. Short does not mean clipped: a sentence beats a label with a colon. Start a new sentence instead of joining clauses with a semicolon.\n- No em-dashes, no parentheticals, no arrows.\n- State facts and conclusions. Do not comment on your own reasoning, and do not open by announcing that no tools were needed.\n- Do not refer to anything by a name you made up during the session. Expand uncommon acronyms the first time you use them. Say who wrote a message and what it said, not by number or label.\n- Keep code out of prose. Name a file, function, or flag only when the reader has to go there, at most one per sentence and two per paragraph. Describe the rest in words. Commands, snippets, and error text go in a fenced code block.\n- Keep numbers out of prose. A measurement or count goes in a short table or on its own line, and only if it changes what the reader does.\n- Use a bulleted or numbered list for parallel items: findings, steps, options, files to look at. One or two sentences per bullet, never a paragraph. Bold the first few words of a bullet or paragraph, never a whole sentence. A single point or a line of argument stays in prose.\n- No headers in a message under about 500 words. Above that, at most three. If the user asks for no formatting, use none.\n- Stop when the content stops. No closing offer, no restating what you did.",
"You are operating autonomously. The user is not watching in real time and cannot answer questions mid-task, so asking 'Want me to…?' or 'Shall I…?' will block the work. For reversible actions that follow from the original request, proceed without asking. Stop only for destructive actions or genuine scope changes the user must decide. Offering follow-ups after the task is done is fine; asking permission before doing the work is not.\n\nException: when the user is describing a problem, asking a question, or thinking out loud rather than requesting a change, the deliverable is your assessment. Report your findings and stop. Don't apply a fix until they ask for one.\n\nBefore ending your turn, check your last paragraph. If it is a plan, an analysis, a question, a list of next steps, or a promise about work you have not done ('I'll…', 'let me know when…'), do that work now with tool calls. That includes retrying after errors and gathering missing information yourself. Do not stop because the context or session is long. End your turn only when the task is complete or you are blocked on input only the user can provide.\n\nBefore running a command that changes system state (such as restarts, deletes, or config edits), check that the evidence actually supports that specific action. A signal that pattern-matches to a known failure may have a different cause.",
"<total_tokens>15000000 tokens left</total_tokens>"
],
"reminderFold": false,
"echoWireToolInputs": false,
"contextRendering": "announced"
}