google / raxLinks
Rax is a Learning-to-Rank library written in JAX.
☆332Updated last week
Alternatives and similar repositories for rax
Users that are interested in rax are comparing it to the libraries listed below
Sorting:
- Distributed skorch on Ray Train☆57Updated 2 years ago
- Weakly Supervised End-to-End Learning (NeurIPS 2021)☆157Updated 2 years ago
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- Pytorch Lightning Distributed Accelerators using Ray☆210Updated last year
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆179Updated 3 weeks ago
- ☆246Updated 3 weeks ago
- A High Level API for Deep Learning in JAX☆475Updated 2 years ago
- Coarse-grained lineage and tracing for machine learning pipelines.☆470Updated 2 years ago
- High performance model preprocessing library on PyTorch☆650Updated last year
- just a bunch of useful embeddings for scikit-learn pipelines☆499Updated 2 months ago
- Functional deep learning☆108Updated 2 years ago
- Merlin Systems provides tools for combining recommendation models with other elements of production recommender systems (like feature sto…☆93Updated 11 months ago
- Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions☆258Updated last year
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆239Updated 2 years ago
- The merlin dataloader lets you rapidly load tabular data for training deep leaning models with TensorFlow, PyTorch or JAX☆418Updated last year
- Super Simple Similarities Service☆148Updated last month
- Task-based datasets, preprocessing, and evaluation for sequence models.☆574Updated 3 weeks ago
- CLU lets you write beautiful training loops in JAX.☆343Updated last month
- Behavioral "black-box" testing for recommender systems☆467Updated last year
- MinT: Minimal Transformer Library and Tutorials☆254Updated 2 years ago
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆261Updated this week
- Drift Detection for your PyTorch Models☆316Updated 2 years ago
- Convert scikit-learn models to PyTorch modules☆161Updated last year
- Recipes are a standard, well supported set of blueprints for machine learning engineers to rapidly train models using the latest research…☆318Updated last week
- A platform for managing machine learning experiments☆853Updated last week
- ☆352Updated last year
- JAX Synergistic Memory Inspector☆172Updated 10 months ago
- Functional tensors for probabilistic programming☆241Updated last year
- A general purpose recommender metrics library for fair evaluation.☆280Updated last year
- Named tensors with first-class dimensions for PyTorch☆329Updated last year