Alternatives and similar repositories for tensorstore

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

• rapidsai / raft
  RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-a…
  ☆903Updated this week
• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆1,053Updated last year
• google-deepmind / dm-haiku
  JAX-based neural network library
  ☆3,053Updated last week
• dmlc / dlpack
  common in-memory tensor structure
  ☆1,019Updated 2 weeks ago
• openxla / xla
  A machine learning compiler for GPUs, CPUs, and ML accelerators
  ☆3,294Updated this week
• google-deepmind / xmanager
  A platform for managing machine learning experiments
  ☆858Updated last month
• google / vizier
  Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.
  ☆1,568Updated this week
• pytorch / data
  A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.
  ☆1,207Updated this week
• NVIDIA / MatX
  An efficient C++17 GPU numerical computing library with Python-like syntax
  ☆1,332Updated this week
• facebookincubator / AITemplate
  AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…
  ☆4,647Updated 2 months ago
• pytorch / torcharrow
  High performance model preprocessing library on PyTorch
  ☆650Updated last year
• rapidsai / rmm
  RAPIDS Memory Manager
  ☆589Updated last week
• patrick-kidger / jaxtyping
  Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
  ☆1,449Updated last month
• google / flax
  Flax is a neural network library for JAX that is designed for flexibility.
  ☆6,634Updated 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 3 months ago
• wjakob / nanobind
  nanobind: tiny and efficient C++/Python bindings
  ☆2,853Updated last week
• google-deepmind / optax
  Optax is a gradient processing and optimization library for JAX.
  ☆1,924Updated this week
• patrick-kidger / equinox
  Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
  ☆2,397Updated last week
• google-deepmind / penzai
  A JAX research toolkit for building, editing, and visualizing neural networks.
  ☆1,796Updated last week
• google-deepmind / dm_pix
  PIX is an image processing library in JAX, for JAX.
  ☆418Updated 3 months ago
• NVIDIA / cccl
  CUDA Core Compute Libraries
  ☆1,711Updated this week
• nv-legate / cupynumeric
  An Aspiring Drop-In Replacement for NumPy at Scale
  ☆902Updated last week
• pytorch / functorch
  functorch is JAX-like composable function transforms for PyTorch.
  ☆1,432Updated this week
• google / autobound
  AutoBound automatically computes upper and lower bounds on functions.
  ☆359Updated last year
• n2cholas / awesome-jax
  JAX - A curated list of resources https://github.com/google/jax
  ☆1,845Updated 4 months ago
• hidet-org / hidet
  An open-source efficient deep learning framework/compiler, written in python.
  ☆704Updated last week
• dblalock / bolt
  10x faster matrix and vector operations
  ☆2,490Updated 2 years ago
• google / grain
  Library for reading and processing ML training data.
  ☆463Updated this week
• nv-legate / legate
  The Foundation for All Legate Libraries
  ☆218Updated last week
• facebookresearch / dietgpu
  GPU implementation of a fast generalized ANS (asymmetric numeral system) entropy encoder and decoder, with extensions for lossless compre…
  ☆338Updated last week