google / rax
Rax is a Learning-to-Rank library written in JAX.
☆324Updated 2 weeks ago
Alternatives and similar repositories for rax:
Users that are interested in rax are comparing it to the libraries listed below
- Weakly Supervised End-to-End Learning (NeurIPS 2021)☆157Updated last year
- Task-based datasets, preprocessing, and evaluation for sequence models.☆566Updated this week
- A High Level API for Deep Learning in JAX☆471Updated 2 years ago
- ☆335Updated 9 months ago
- Distributed skorch on Ray Train☆57Updated 2 years ago
- A platform for managing machine learning experiments☆832Updated last month
- The merlin dataloader lets you rapidly load tabular data for training deep leaning models with TensorFlow, PyTorch or JAX☆413Updated 9 months ago
- Robustness Gym is an evaluation toolkit for machine learning.☆442Updated 2 years ago
- CLU lets you write beautiful training loops in JAX.☆329Updated this week
- Pytorch Lightning Distributed Accelerators using Ray☆211Updated last year
- High performance model preprocessing library on PyTorch☆651Updated 9 months ago
- ☆197Updated last week
- just a bunch of useful embeddings for scikit-learn pipelines☆474Updated 3 weeks ago
- MinT: Minimal Transformer Library and Tutorials☆251Updated 2 years ago
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆237Updated last year
- Behavioral "black-box" testing for recommender systems☆464Updated last year
- A Jax-based library for designing and training transformer models from scratch.☆280Updated 4 months ago
- A library for distributed ML training with PyTorch☆366Updated 2 years ago
- Coarse-grained lineage and tracing for machine learning pipelines.☆467Updated 2 years ago
- Merlin Systems provides tools for combining recommendation models with other elements of production recommender systems (like feature sto…☆90Updated 7 months ago
- Check if you have training samples in your test set☆64Updated 2 years ago
- ☆340Updated this week
- Merlin Models is a collection of deep learning recommender system model reference implementations☆266Updated 8 months ago
- ☆110Updated last week
- Named tensors with first-class dimensions for PyTorch☆322Updated last year
- Drift Detection for your PyTorch Models☆312Updated 2 years ago
- ☆806Updated this week
- Functional deep learning☆108Updated 2 years ago
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆175Updated 2 years ago