Alternatives and similar repositories for tensorstore:

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

• pytorch / torcharrow
  High performance model preprocessing library on PyTorch
  ☆650Updated last year
• google / vizier
  Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.
  ☆1,541Updated 2 weeks ago
• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆1,037Updated 11 months ago
• pytorch / data
  A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.
  ☆1,177Updated this week
• rapidsai / raft
  RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-a…
  ☆860Updated this week
• nv-legate / cupynumeric
  An Aspiring Drop-In Replacement for NumPy at Scale
  ☆854Updated this week
• aesara-devs / aesara
  Aesara is a Python library for defining, optimizing, and efficiently evaluating mathematical expressions involving multi-dimensional arra…
  ☆1,200Updated 4 months ago
• dmlc / dlpack
  common in-memory tensor structure
  ☆968Updated this week
• NVIDIA / aistore
  AIStore: scalable storage for AI applications
  ☆1,441Updated this week
• google / paxml
  Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…
  ☆485Updated last week
• google-deepmind / dm-haiku
  JAX-based neural network library
  ☆2,996Updated 2 weeks ago
• NVIDIA / MatX
  An efficient C++17 GPU numerical computing library with Python-like syntax
  ☆1,303Updated this week
• google / fiddle
  ☆345Updated last week
• openxla / xla
  A machine learning compiler for GPUs, CPUs, and ML accelerators
  ☆3,045Updated this week
• dblalock / bolt
  10x faster matrix and vector operations
  ☆2,481Updated 2 years ago
• pytorch / torchx
  TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and sup…
  ☆354Updated this week
• google-deepmind / xmanager
  A platform for managing machine learning experiments
  ☆841Updated last week
• rapidsai / rmm
  RAPIDS Memory Manager
  ☆559Updated this week
• google-deepmind / optax
  Optax is a gradient processing and optimization library for JAX.
  ☆1,839Updated 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,560Updated last year
• rapidsai / kvikio
  KvikIO - High Performance File IO
  ☆198Updated this week
• google-deepmind / chex
  ☆841Updated 2 weeks ago
• facebookincubator / AITemplate
  AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…
  ☆4,622Updated 3 months ago
• poets-ai / elegy
  A High Level API for Deep Learning in JAX
  ☆473Updated 2 years ago
• zarr-developers / zarr-python
  An implementation of chunked, compressed, N-dimensional arrays for Python.
  ☆1,647Updated last week
• NVIDIA / cuCollections
  ☆530Updated last week
• facebookresearch / moolib
  A library for distributed ML training with PyTorch
  ☆366Updated 2 years ago
• pytorch / functorch
  functorch is JAX-like composable function transforms for PyTorch.
  ☆1,416Updated this week
• google / grain
  Library for reading and processing ML training data.
  ☆414Updated this week
• google-deepmind / penzai
  A JAX research toolkit for building, editing, and visualizing neural networks.
  ☆1,748Updated 3 months ago