Alternatives and similar repositories for tensorstore:

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

• google / vizier
  Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.
  ☆1,553Updated last week
• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆1,040Updated last year
• rapidsai / raft
  RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-a…
  ☆870Updated this week
• google-deepmind / dm-haiku
  JAX-based neural network library
  ☆3,013Updated this week
• dmlc / dlpack
  common in-memory tensor structure
  ☆982Updated 2 weeks ago
• pytorch / torcharrow
  High performance model preprocessing library on PyTorch
  ☆650Updated last year
• openxla / xla
  A machine learning compiler for GPUs, CPUs, and ML accelerators
  ☆3,112Updated this week
• tuplex / tuplex
  Tuplex is a parallel big data processing framework that runs data science pipelines written in Python at the speed of compiled code. Tupl…
  ☆809Updated last month
• rapidsai / rmm
  RAPIDS Memory Manager
  ☆573Updated this week
• pytorch / data
  A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.
  ☆1,190Updated this week
• ELS-RD / kernl
  Kernl lets you run PyTorch transformer models several times faster on GPU with a single line of code, and is designed to be easily hackab…
  ☆1,564Updated last year
• facebookincubator / AITemplate
  AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…
  ☆4,630Updated 3 weeks ago
• pytorch / FBGEMM
  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  ☆1,303Updated this week
• NVIDIA / MatX
  An efficient C++17 GPU numerical computing library with Python-like syntax
  ☆1,313Updated this week
• facebookresearch / moolib
  A library for distributed ML training with PyTorch
  ☆366Updated 2 years ago
• google / ml_collections
  ML Collections is a library of Python Collections designed for ML use cases.
  ☆945Updated last week
• google-deepmind / xmanager
  A platform for managing machine learning experiments
  ☆846Updated this week
• Lightning-AI / lightning-thunder
  Thunder gives you PyTorch models superpowers for training and inference. Unlock out-of-the-box optimizations for performance, memory and …
  ☆1,328Updated this week
• google / fiddle
  ☆347Updated this week
• zarr-developers / zarr-python
  An implementation of chunked, compressed, N-dimensional arrays for Python.
  ☆1,677Updated this week
• google-deepmind / optax
  Optax is a gradient processing and optimization library for JAX.
  ☆1,868Updated last week
• google / flax
  Flax is a neural network library for JAX that is designed for flexibility.
  ☆6,506Updated this week
• facebookincubator / gloo
  Collective communications library with various primitives for multi-machine training.
  ☆1,289Updated this week
• omry / omegaconf
  Flexible Python configuration system. The last one you will ever need.
  ☆2,121Updated 3 months ago
• pytorch / functorch
  functorch is JAX-like composable function transforms for PyTorch.
  ☆1,422Updated this week
• google / grain
  Library for reading and processing ML training data.
  ☆432Updated this week
• tensorflow / runtime
  A performant and modular runtime for TensorFlow
  ☆759Updated last week
• rapidsai / kvikio
  KvikIO - High Performance File IO
  ☆206Updated this week
• pytorch / tensordict
  TensorDict is a pytorch dedicated tensor container.
  ☆911Updated this week
• facebookresearch / dietgpu
  GPU implementation of a fast generalized ANS (asymmetric numeral system) entropy encoder and decoder, with extensions for lossless compre…
  ☆325Updated last month