google / rax
Rax is a Learning-to-Rank library written in JAX.
☆327Updated 2 weeks ago
Alternatives and similar repositories for rax:
Users that are interested in rax are comparing it to the libraries listed below
- Task-based datasets, preprocessing, and evaluation for sequence models.☆571Updated this week
- A High Level API for Deep Learning in JAX☆473Updated 2 years ago
- High performance model preprocessing library on PyTorch☆650Updated 11 months ago
- ☆345Updated this week
- Distributed skorch on Ray Train☆57Updated 2 years ago
- ☆343Updated 11 months ago
- Named tensors with first-class dimensions for PyTorch☆321Updated last year
- Pytorch Lightning Distributed Accelerators using Ray☆211Updated last year
- just a bunch of useful embeddings for scikit-learn pipelines☆486Updated 2 months ago
- Weakly Supervised End-to-End Learning (NeurIPS 2021)☆156Updated 2 years ago
- Coarse-grained lineage and tracing for machine learning pipelines.☆467Updated 2 years ago
- The merlin dataloader lets you rapidly load tabular data for training deep leaning models with TensorFlow, PyTorch or JAX☆417Updated 11 months ago
- A platform for managing machine learning experiments☆840Updated this week
- MinT: Minimal Transformer Library and Tutorials☆253Updated 2 years ago
- jax-triton contains integrations between JAX and OpenAI Triton☆386Updated last week
- ☆832Updated this week
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆236Updated last year
- CLU lets you write beautiful training loops in JAX.☆335Updated 2 weeks ago
- Robustness Gym is an evaluation toolkit for machine learning.☆442Updated 2 years ago
- A Jax-based library for designing and training transformer models from scratch.☆282Updated 6 months ago
- Merlin Systems provides tools for combining recommendation models with other elements of production recommender systems (like feature sto…☆91Updated 9 months ago
- Hopular: Modern Hopfield Networks for Tabular Data☆310Updated 2 years ago
- A Pytree Module system for Deep Learning in JAX☆213Updated 2 years ago
- skops is a Python library helping you share your scikit-learn based models and put them in production☆467Updated 2 weeks ago
- Convert scikit-learn models to PyTorch modules☆161Updated 10 months ago
- A Python library that enables ML teams to share, load, and transform data in a collaborative, flexible, and efficient way☆281Updated this week
- Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions☆257Updated last year
- ☆245Updated 5 months ago
- ☆153Updated 4 years ago
- Mistral: A strong, northwesterly wind: Framework for transparent and accessible large-scale language model training, built with Hugging F…☆569Updated last year