google / raxLinks
Rax is a Learning-to-Rank library written in JAX.
☆336Updated this week
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)☆156Updated 2 years ago
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆298Updated last year
- A High Level API for Deep Learning in JAX☆476Updated 3 years ago
- Functional deep learning☆108Updated 3 years ago
- Merlin Systems provides tools for combining recommendation models with other elements of production recommender systems (like feature sto…☆94Updated last year
- Pytorch Lightning Distributed Accelerators using Ray☆215Updated 2 years ago
- MinT: Minimal Transformer Library and Tutorials☆260Updated 3 years ago
- Coarse-grained lineage and tracing for machine learning pipelines.☆471Updated 3 years ago
- Behavioral "black-box" testing for recommender systems☆468Updated 2 years ago
- The merlin dataloader lets you rapidly load tabular data for training deep leaning models with TensorFlow, PyTorch or JAX☆423Updated last year
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- Hopular: Modern Hopfield Networks for Tabular Data☆313Updated 3 years ago
- Distributed skorch on Ray Train☆58Updated 3 years ago
- High performance model preprocessing library on PyTorch☆646Updated last year
- A library that allows for inference on probabilistic models☆273Updated 2 years ago
- Drift Detection for your PyTorch Models☆324Updated 3 years ago
- Task-based datasets, preprocessing, and evaluation for sequence models.☆594Updated this week
- A platform for managing machine learning experiments☆894Updated last month
- Check if you have training samples in your test set☆64Updated 3 years ago
- ☆367Updated last year
- Robustness Gym is an evaluation toolkit for machine learning.☆445Updated 3 years ago
- Docs☆143Updated last year
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆242Updated 2 years ago
- ☆374Updated this week
- just a bunch of useful embeddings for scikit-learn pipelines☆520Updated 4 months ago
- A python package for benchmarking interpretability techniques on Transformers.☆215Updated last year
- ☆106Updated last year
- A general purpose recommender metrics library for fair evaluation.☆277Updated 2 years ago
- Recommendations at "Reasonable Scale": joining dataOps with recSys through dbt, Merlin and Metaflow☆241Updated 2 years ago
- Super Simple Similarities Service☆155Updated 9 months ago