Alternatives and similar repositories for tensorstore

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

• openxla / xla
  A machine learning compiler for GPUs, CPUs, and ML accelerators
  ☆3,219Updated this week
• pytorch / torcharrow
  High performance model preprocessing library on PyTorch
  ☆650Updated last year
• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆1,049Updated last year
• google / vizier
  Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.
  ☆1,562Updated last month
• google-deepmind / xmanager
  A platform for managing machine learning experiments
  ☆857Updated 2 weeks ago
• pytorch / functorch
  functorch is JAX-like composable function transforms for PyTorch.
  ☆1,431Updated this week
• rapidsai / raft
  RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-a…
  ☆891Updated this week
• google-deepmind / dm-haiku
  JAX-based neural network library
  ☆3,041Updated last week
• google-deepmind / penzai
  A JAX research toolkit for building, editing, and visualizing neural networks.
  ☆1,784Updated last month
• google / fiddle
  ☆348Updated 3 weeks ago
• nv-legate / cupynumeric
  An Aspiring Drop-In Replacement for NumPy at Scale
  ☆898Updated this week
• NVIDIA / MatX
  An efficient C++17 GPU numerical computing library with Python-like syntax
  ☆1,325Updated 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,645Updated 2 months ago
• dblalock / bolt
  10x faster matrix and vector operations
  ☆2,488Updated 2 years ago
• google / paxml
  Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…
  ☆499Updated 2 weeks ago
• google / flax
  Flax is a neural network library for JAX that is designed for flexibility.
  ☆6,605Updated this week
• google / ml_collections
  ML Collections is a library of Python Collections designed for ML use cases.
  ☆955Updated last month
• google-deepmind / optax
  Optax is a gradient processing and optimization library for JAX.
  ☆1,908Updated this week
• pytorch / FBGEMM
  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  ☆1,348Updated 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 2 months ago
• dmlc / dlpack
  common in-memory tensor structure
  ☆1,002Updated 3 weeks ago
• pytorch / tensordict
  TensorDict is a pytorch dedicated tensor container.
  ☆927Updated this week
• google-deepmind / dm_pix
  PIX is an image processing library in JAX, for JAX.
  ☆417Updated 3 months ago
• rapidsai / rmm
  RAPIDS Memory Manager
  ☆586Updated last week
• aesara-devs / aesara
  Aesara is a Python library for defining, optimizing, and efficiently evaluating mathematical expressions involving multi-dimensional arra…
  ☆1,207Updated 6 months ago
• google-deepmind / chex
  ☆856Updated this week
• tensorflow / mesh
  Mesh TensorFlow: Model Parallelism Made Easier
  ☆1,608Updated last year
• nv-legate / legate
  The Foundation for All Legate Libraries
  ☆218Updated this week
• hidet-org / hidet
  An open-source efficient deep learning framework/compiler, written in python.
  ☆700Updated this week
• patrick-kidger / equinox
  Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
  ☆2,381Updated 2 weeks ago