Alternatives and similar repositories for rax:

Users that are interested in rax are comparing it to the libraries listed below

• autonlab / weasel
  Weakly Supervised End-to-End Learning (NeurIPS 2021)
  ☆157Updated last year
• google / seqio
  Task-based datasets, preprocessing, and evaluation for sequence models.
  ☆568Updated this week
• google / flaxformer
  ☆337Updated 10 months ago
• NVIDIA-Merlin / dataloader
  The merlin dataloader lets you rapidly load tabular data for training deep leaning models with TensorFlow, PyTorch or JAX
  ☆414Updated 10 months ago
• poets-ai / elegy
  A High Level API for Deep Learning in JAX
  ☆472Updated 2 years ago
• google-deepmind / chex
  ☆822Updated last week
• google / fiddle
  ☆340Updated 2 weeks ago
• loglabs / mltrace
  Coarse-grained lineage and tracing for machine learning pipelines.
  ☆467Updated 2 years ago
• google / CommonLoopUtils
  CLU lets you write beautiful training loops in JAX.
  ☆333Updated 2 weeks ago
• Yard1 / ray-skorch
  Distributed skorch on Ray Train
  ☆57Updated 2 years ago
• HMUNACHI / nanodl
  A Jax-based library for designing and training transformer models from scratch.
  ☆281Updated 5 months ago
• google-deepmind / xmanager
  A platform for managing machine learning experiments
  ☆838Updated 3 weeks ago
• explosion / floret
  🌸 fastText + Bloom embeddings for compact, full-coverage vectors with spaCy
  ☆307Updated last year
• google / orbax
  Orbax provides common checkpointing and persistence utilities for JAX users
  ☆338Updated this week
• facebookresearch / moolib
  A library for distributed ML training with PyTorch
  ☆366Updated 2 years ago
• koaning / embetter
  just a bunch of useful embeddings for scikit-learn pipelines
  ☆480Updated 3 weeks ago
• NVIDIA-Merlin / systems
  Merlin Systems provides tools for combining recommendation models with other elements of production recommender systems (like feature sto…
  ☆91Updated 8 months ago
• google / praxis
  ☆183Updated 2 weeks ago
• cgarciae / treex
  A Pytree Module system for Deep Learning in JAX
  ☆213Updated last year
• google-deepmind / synjax
  ☆244Updated 4 months ago
• lf1-io / padl
  Functional deep learning
  ☆108Updated 2 years ago
• google-deepmind / tf2jax
  ☆111Updated 2 weeks ago
• stanford-crfm / haliax
  Named Tensors for Legible Deep Learning in JAX
  ☆161Updated this week
• kingoflolz / swarm-jax
  Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes
  ☆237Updated last year
• pytorch / torcharrow
  High performance model preprocessing library on PyTorch
  ☆651Updated 10 months ago
• NVIDIA-Merlin / models
  Merlin Models is a collection of deep learning recommender system model reference implementations
  ☆269Updated 9 months ago
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆321Updated last year
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆191Updated 2 weeks ago
• ray-project / ray_lightning
  Pytorch Lightning Distributed Accelerators using Ray
  ☆211Updated last year
• TorchDrift / TorchDrift
  Drift Detection for your PyTorch Models
  ☆315Updated 2 years ago