Alternatives and similar repositories for tensorstore:

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

• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆1,035Updated 10 months ago
• pytorch / torcharrow
  High performance model preprocessing library on PyTorch
  ☆651Updated 11 months ago
• openxla / xla
  A machine learning compiler for GPUs, CPUs, and ML accelerators
  ☆3,017Updated this week
• dmlc / dlpack
  common in-memory tensor structure
  ☆959Updated 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…
  ☆810Updated this week
• rapidsai / raft
  RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-a…
  ☆852Updated this week
• google / flax
  Flax is a neural network library for JAX that is designed for flexibility.
  ☆6,400Updated this week
• facebookresearch / fairscale
  PyTorch extensions for high performance and large scale training.
  ☆3,272Updated 2 months ago
• google / vizier
  Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.
  ☆1,537Updated 3 weeks ago
• 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,559Updated last year
• pytorch / kineto
  A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
  ☆778Updated this week
• google-deepmind / dm-haiku
  JAX-based neural network library
  ☆2,981Updated last month
• pytorch / data
  A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.
  ☆1,169Updated this week
• huggingface / safetensors
  Simple, safe way to store and distribute tensors
  ☆3,161Updated last week
• NVIDIA / MatX
  An efficient C++17 GPU numerical computing library with Python-like syntax
  ☆1,276Updated this week
• rapidsai / rmm
  RAPIDS Memory Manager
  ☆551Updated this week
• hidet-org / hidet
  An open-source efficient deep learning framework/compiler, written in python.
  ☆689Updated 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,615Updated 3 months ago
• google-deepmind / xmanager
  A platform for managing machine learning experiments
  ☆839Updated 3 weeks 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…
  ☆349Updated this week
• google / fiddle
  ☆344Updated last month
• zarr-developers / zarr-python
  An implementation of chunked, compressed, N-dimensional arrays for Python.
  ☆1,632Updated this week
• google-deepmind / alphatensor
  ☆2,726Updated 10 months ago
• pytorch / functorch
  functorch is JAX-like composable function transforms for PyTorch.
  ☆1,415Updated this week
• TorchStudio / torchstudio
  IDE for PyTorch and its ecosystem
  ☆384Updated 10 months ago
• Lightning-AI / litdata
  Transform datasets at scale. Optimize datasets for fast AI model training.
  ☆421Updated this week
• patrick-kidger / equinox
  Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
  ☆2,255Updated 2 weeks ago
• webdataset / webdataset
  A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.
  ☆2,494Updated last month
• google-deepmind / penzai
  A JAX research toolkit for building, editing, and visualizing neural networks.
  ☆1,742Updated 2 months ago
• google / ml_collections
  ML Collections is a library of Python Collections designed for ML use cases.
  ☆924Updated last month