google / raxLinks
Rax is a Learning-to-Rank library written in JAX.
☆331Updated last month
Alternatives and similar repositories for rax
Users that are interested in rax are comparing it to the libraries listed below
Sorting:
- The merlin dataloader lets you rapidly load tabular data for training deep leaning models with TensorFlow, PyTorch or JAX☆418Updated last year
- High performance model preprocessing library on PyTorch☆650Updated last year
- Distributed skorch on Ray Train☆57Updated 2 years ago
- Weakly Supervised End-to-End Learning (NeurIPS 2021)☆157Updated 2 years ago
- ☆350Updated last month
- Drift Detection for your PyTorch Models☆317Updated 2 years ago
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆287Updated 9 months ago
- Merlin Systems provides tools for combining recommendation models with other elements of production recommender systems (like feature sto…☆93Updated last year
- A High Level API for Deep Learning in JAX☆475Updated 2 years ago
- Task-based datasets, preprocessing, and evaluation for sequence models.☆576Updated last month
- Merlin Models is a collection of deep learning recommender system model reference implementations☆281Updated last year
- ☆861Updated 2 weeks ago
- Provides a common interface to many IR ranking datasets.☆358Updated last month
- Named tensors with first-class dimensions for PyTorch☆331Updated 2 years ago
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- Pytorch Lightning Distributed Accelerators using Ray☆211Updated last year
- Check if you have training samples in your test set☆64Updated 3 years ago
- MinT: Minimal Transformer Library and Tutorials☆255Updated 2 years ago
- Functional deep learning☆108Updated 2 years ago
- ☆578Updated last month
- A library for distributed ML training with PyTorch☆366Updated 2 years ago
- skops is a Python library helping you share your scikit-learn based models and put them in production☆490Updated last week
- ☆199Updated last month
- Recipes are a standard, well supported set of blueprints for machine learning engineers to rapidly train models using the latest research…☆319Updated last week
- Behavioral "black-box" testing for recommender systems☆468Updated last year
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆239Updated 2 years ago
- ☆355Updated last year
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆179Updated last month
- ☆246Updated last month
- just a bunch of useful embeddings for scikit-learn pipelines☆500Updated 3 months ago