google / rax
Rax is a Learning-to-Rank library written in JAX.
☆328Updated last month
Alternatives and similar repositories for rax:
Users that are interested in rax are comparing it to the libraries listed below
- Distributed skorch on Ray Train☆57Updated 2 years ago
- Functional deep learning☆108Updated 2 years ago
- Pytorch Lightning Distributed Accelerators using Ray☆210Updated last year
- Weakly Supervised End-to-End Learning (NeurIPS 2021)☆156Updated 2 years ago
- High performance model preprocessing library on PyTorch☆650Updated last year
- A High Level API for Deep Learning in JAX☆474Updated 2 years ago
- Merlin Systems provides tools for combining recommendation models with other elements of production recommender systems (like feature sto…☆91Updated 10 months ago
- The merlin dataloader lets you rapidly load tabular data for training deep leaning models with TensorFlow, PyTorch or JAX☆417Updated 11 months ago
- ☆346Updated this week
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- ☆346Updated last year
- A Jax-based library for designing and training transformer models from scratch.☆285Updated 7 months ago
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆238Updated last year
- A library for distributed ML training with PyTorch☆366Updated 2 years ago
- Drift Detection for your PyTorch Models☆316Updated 2 years ago
- MinT: Minimal Transformer Library and Tutorials☆253Updated 2 years ago
- Merlin Models is a collection of deep learning recommender system model reference implementations☆274Updated 11 months ago
- Named tensors with first-class dimensions for PyTorch☆320Updated last year
- Recipes are a standard, well supported set of blueprints for machine learning engineers to rapidly train models using the latest research…☆312Updated last week
- Check if you have training samples in your test set☆64Updated 2 years ago
- Task-based datasets, preprocessing, and evaluation for sequence models.☆571Updated 3 weeks ago
- A Python library that enables ML teams to share, load, and transform data in a collaborative, flexible, and efficient way☆281Updated 3 weeks ago
- Library for reading and processing ML training data.☆425Updated this week
- spock is a framework that helps manage complex parameter configurations during research and development of Python applications☆132Updated last year
- Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions☆258Updated last year
- ☆199Updated last week
- Coarse-grained lineage and tracing for machine learning pipelines.☆468Updated 2 years ago
- just a bunch of useful embeddings for scikit-learn pipelines☆495Updated 3 weeks ago
- Hopular: Modern Hopfield Networks for Tabular Data☆311Updated 2 years ago
- CLU lets you write beautiful training loops in JAX.☆335Updated this week