google / rax
Rax is a Learning-to-Rank library written in JAX.
☆319Updated 3 weeks ago
Related projects ⓘ
Alternatives and complementary repositories for rax
- Weakly Supervised End-to-End Learning (NeurIPS 2021)☆153Updated last year
- Distributed skorch on Ray Train☆57Updated 2 years ago
- High performance model preprocessing library on PyTorch☆649Updated 7 months ago
- Pytorch Lightning Distributed Accelerators using Ray☆210Updated last year
- Coarse-grained lineage and tracing for machine learning pipelines.☆468Updated 2 years ago
- Merlin Systems provides tools for combining recommendation models with other elements of production recommender systems (like feature sto…☆90Updated 5 months ago
- Robustness Gym is an evaluation toolkit for machine learning.☆441Updated 2 years ago
- ☆328Updated this week
- A Python library that enables ML teams to share, load, and transform data in a collaborative, flexible, and efficient way☆282Updated this week
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆175Updated 2 years ago
- Merlin Models is a collection of deep learning recommender system model reference implementations☆262Updated 6 months ago
- Functional deep learning☆106Updated 2 years ago
- Task-based datasets, preprocessing, and evaluation for sequence models.☆561Updated this week
- just a bunch of useful embeddings☆467Updated 2 months ago
- A High Level API for Deep Learning in JAX☆470Updated last year
- A Jax-based library for designing and training transformer models from scratch.☆276Updated 2 months ago
- CLU lets you write beautiful training loops in JAX.☆321Updated last week
- Behavioral "black-box" testing for recommender systems☆459Updated last year
- A python package for benchmarking interpretability techniques on Transformers.☆212Updated last month
- A Pytree Module system for Deep Learning in JAX☆214Updated last year
- ☆334Updated 7 months ago
- Recommendations at "Reasonable Scale": joining dataOps with recSys through dbt, Merlin and Metaflow☆230Updated last year
- A platform for managing machine learning experiments☆818Updated 3 months ago
- The merlin dataloader lets you rapidly load tabular data for training deep leaning models with TensorFlow, PyTorch or JAX☆408Updated 7 months ago
- Recipes are a standard, well supported set of blueprints for machine learning engineers to rapidly train models using the latest research…☆295Updated this week
- ☆787Updated this week
- deep learning with pytorch lightning☆0Updated 3 weeks ago
- Drift Detection for your PyTorch Models☆312Updated 2 years ago
- A library that allows for inference on probabilistic models☆265Updated last year