google / raxLinks
Rax is a Learning-to-Rank library written in JAX.
☆330Updated last month
Alternatives and similar repositories for rax
Users that are interested in rax are comparing it to the libraries listed below
Sorting:
- Weakly Supervised End-to-End Learning (NeurIPS 2021)☆157Updated 2 years ago
- A High Level API for Deep Learning in JAX☆474Updated 2 years ago
- High performance model preprocessing library on PyTorch☆649Updated last year
- The merlin dataloader lets you rapidly load tabular data for training deep leaning models with TensorFlow, PyTorch or JAX☆419Updated last year
- Task-based datasets, preprocessing, and evaluation for sequence models.☆583Updated 2 months ago
- Hopular: Modern Hopfield Networks for Tabular Data☆311Updated 3 years ago
- MinT: Minimal Transformer Library and Tutorials☆256Updated 2 years ago
- Pytorch Lightning Distributed Accelerators using Ray☆211Updated last year
- Coarse-grained lineage and tracing for machine learning pipelines.☆469Updated 2 years ago
- ☆358Updated last year
- Merlin Systems provides tools for combining recommendation models with other elements of production recommender systems (like feature sto…☆94Updated last year
- A library that allows for inference on probabilistic models☆272Updated last year
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆290Updated 10 months ago
- Distributed skorch on Ray Train☆57Updated 2 years ago
- ☆350Updated this week
- Drift Detection for your PyTorch Models☆317Updated 2 years ago
- A platform for managing machine learning experiments☆859Updated last month
- Behavioral "black-box" testing for recommender systems☆467Updated last year
- Docs☆144Updated 7 months ago
- Check if you have training samples in your test set☆64Updated 3 years ago
- deep learning with pytorch lightning☆1Updated 8 months ago
- Organize your experiments into discrete steps that can be cached and reused throughout the lifetime of your research project.☆563Updated last year
- A word2vec negative sampling implementation with correct CBOW update.☆261Updated 3 years ago
- Merlin Models is a collection of deep learning recommender system model reference implementations☆283Updated last year
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- 🌸 fastText + Bloom embeddings for compact, full-coverage vectors with spaCy☆315Updated 2 months ago
- Robustness Gym is an evaluation toolkit for machine learning.☆442Updated 3 years ago
- Super Simple Similarities Service☆150Updated 3 months ago
- A python package for benchmarking interpretability techniques on Transformers.☆213Updated 9 months ago
- Named tensors with first-class dimensions for PyTorch☆332Updated 2 years ago