workflowstacks
ai-agent

Rax: Boost Ranking Models with JAX‑Powered Learning‑to‑Rank

Rax lets founders train and deploy learning‑to‑rank models in Python using JAX, accelerating search and recommendation pipelines.

From the repo · “Rax is a Learning-to-Rank library written in JAX.”

342 stars13 forksPythonGuide quality 9/10 · reviewed Oct 10Updated 10/5/2026100% free · open source
What it does

Rax is a library that helps you train machine learning models to rank items in a list, such as search results or product recommendations.

When to use it
  • •Building a search engine that needs to rank results by relevance
  • •Creating a product recommendation system that suggests items based on user behavior
  • •Developing a content ranking system for a news feed or social media platform
Ready-to-paste prompt
python examples/train_model.py --dataset=example_data --model=ranking_model
Heads up: Rax requires JAX to be installed, which can be challenging to install on certain platforms, especially those without GPU support, so ensure you have a compatible environment before installing Rax.
Saves to your device
Use with Claude
New

Skip the builder — one click puts this in Claude, Cursor, Antigravity and more.

✅ Light setup

Installs with a command or two; your AI agent can do it for you.

Try it instantly — no install
Claude Code
mkdir -p ~/.claude/skills/rax && curl -fsSL https://workflowstacks.com/api/skills/rax/claude-skill -o ~/.claude/skills/rax/SKILL.md
Open in another AI app

Opens the app with this repo with the prompt ready to go — no copy-paste needed.

Connect the whole catalog (MCP)
claude mcp add --transport http workflowstacks https://workflowstacks.com/api/mcp

Adds a WorkflowStacks connector to Claude Code: search and load any skill here by chatting.

How Rax: Boost Ranking Models with JAX‑Powered Learning‑to‑Rank works
Codeflow
Free to inspect

Rax: Boost Ranking Models with JAX‑Powered Learning‑to‑Rank is a medium Python project (~8.1k lines across 33 code files). Setup is light: installs like a normal app. Reading the code is optional. Last commit a month ago, Apache-2.0 license, has a test suite.

Size
Medium codebase
~8.1k lines · 33 code files · ~1 h to skim
Setup
One-command install
Installs like a normal app. Reading the code is optional.
Runs on
Python
No API keys detected
Python 100%
Where to start reading
  1. 1
    README.md
    Start here — what it does and how to install it
  2. 2
    rax/__init__.py
    Inside rax/ — the main logic begins here
  3. 3
    examples/__init__.py
    A worked example — copy this to get going
What's in each folder
rax/Core code — the actual logic17 files
examples/Examples you can copy20 files
docs/Documentation10 files
requirements/Folder4 files
.github/CI / automation (GitHub Actions)2 files
READMEHas testsDocumentedCI checksExamples includedApache-2.0 licenseUpdated this month
Quick Actions
Details
Creator
google
Language
Python
Category
ai-agent
Published
1/13/2022

Are you the creator of this tool? and earn 85% of every sale.

Show it off in your README:

Featured on WorkflowStacks
[![Featured on WorkflowStacks](https://workflowstacks.com/api/badge/rax.svg)](https://workflowstacks.com/skills/rax?utm_source=github&utm_medium=badge)
🔥 Hot this week, in your inbox
Skills like this, every Monday.

The five fastest-growing open-source AI skills, ranked by GitHub star growth. One email a week, unsubscribe anytime.

Ready-to-run pack · one-time $29

Rank Tracker

A weekly email of where you actually rank on Google.

Includes the tested workflow and a written setup playbook. Needs: SerpAPI key (free tier).

See what you get

Related skills

More ai-agent tools founders pair with this one.

ai-agent★ 303,229
superpowers
An agentic skills framework & software development methodology that works.
ai-agent★ 277,154
Track GitHub Stars with 996.ICU
Get a popular GitHub star counter for your startup, backed by 276k+ GitHub stars, ideal for founders.
ai-agent★ 253,101
Linux – 251k+ GitHub stars for building custom kernel solutions
Get the complete Linux kernel source in C, perfect for founders needing deep system control or tailored OS components.
ai-agent★ 8,684
Jesse: Python Crypto Trading Bot for Faster Market Moves
Leverage Jesse’s advanced algorithmic trading framework to automate Bitcoin and crypto strategies—perfect for fintech founders looking to scale trading operations. 8.6k+ GitHub stars.
ai-agent★ 45,037
jan
Jan is an open source alternative to ChatGPT that runs 100% offline on your computer.
ai-agent★ 14,182
semantica: Build Trustworthy AI with Graph‑Native Context
Leverage semantica’s Python‑based graph infrastructure to add transparent, accountable memory to LLM agents. Ideal for founders building explainable AI products.