Rax is a Learning-to-Rank library written in JAX.
☆337Feb 13, 2026Updated 2 weeks ago
Alternatives and similar repositories for rax
Users that are interested in rax are comparing it to the libraries listed below
Sorting:
- CLU lets you write beautiful training loops in JAX.☆366Feb 20, 2026Updated last week
- Turn SymPy expressions into trainable JAX expressions.☆362Feb 4, 2026Updated 3 weeks ago
- (Personalized) Page-Rank computation using PyTorch☆93Nov 30, 2022Updated 3 years ago
- Search Engines with Autoregressive Language models☆295Apr 4, 2023Updated 2 years ago
- ☆2,946Jan 15, 2026Updated last month
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆549Updated this week
- A Graph Neural Network Library in Jax☆1,464Mar 18, 2024Updated last year
- ☆192Feb 16, 2026Updated last week
- Mathematical operations for JAX pytrees☆209Dec 5, 2024Updated last year
- T5Patches is a set of tools for fast and targeted editing of generative language models built with T5X.☆12May 31, 2024Updated last year
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆1,027Dec 17, 2025Updated 2 months ago
- PIX is an image processing library in JAX, for JAX.☆434Mar 6, 2025Updated 11 months ago
- Flax is a neural network library for JAX that is designed for flexibility.☆7,082Updated this week
- JAX-based neural network library☆3,191Updated this week
- This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.☆14Dec 21, 2021Updated 4 years ago
- Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.☆1,629Feb 17, 2026Updated last week
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,784Feb 21, 2026Updated last week
- ☆925Jan 29, 2026Updated last month
- Composable metric reporters in Python.☆14Jun 6, 2024Updated last year
- A JAX implementation of stochastic addition.☆14Aug 15, 2022Updated 3 years ago
- ☆22Mar 16, 2025Updated 11 months ago
- allRank is a framework for training learning-to-rank neural models based on PyTorch.☆993Aug 6, 2024Updated last year
- Task-based datasets, preprocessing, and evaluation for sequence models.☆594Feb 3, 2026Updated 3 weeks ago
- ☆367Apr 12, 2024Updated last year
- Unbiased Learning To Rank Algorithms (ULTRA)☆94Feb 1, 2023Updated 3 years ago
- BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.☆1,020Feb 3, 2026Updated 3 weeks ago
- Orbax provides common checkpointing and persistence utilities for JAX users☆482Updated this week
- ☆63Sep 10, 2020Updated 5 years ago
- Library for reading and processing ML training data.☆684Feb 21, 2026Updated last week
- jax-triton contains integrations between JAX and OpenAI Triton☆439Updated this week
- Optax is a gradient processing and optimization library for JAX.☆2,193Updated this week
- A Pytree Module system for Deep Learning in JAX☆213Feb 26, 2023Updated 3 years ago
- Learning to Rank in TensorFlow☆2,779Mar 18, 2024Updated last year
- Code to statistically up-weight conversion values of consenting customers to feed up to 100% of the factual conversion values back into G…☆27Jan 12, 2026Updated last month
- Evolution Strategies in JAX 🦎☆729Sep 20, 2025Updated 5 months ago
- For optimization algorithm research and development.☆558Updated this week
- Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.☆2,610Updated this week
- A Python framework for performing information retrieval experiments, building on http://terrier.org/☆495Updated this week
- ☆164Dec 13, 2023Updated 2 years ago