Library for reading and writing large multi-dimensional arrays.
☆1,495Updated this week
Alternatives and similar repositories for tensorstore
Users that are interested in tensorstore are comparing it to the libraries listed below
Sorting:
- An implementation of chunked, compressed, N-dimensional arrays for Python.☆1,919Feb 21, 2026Updated last week
- AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…☆4,706Jan 12, 2026Updated last month
- Not HDF5☆198Updated this week
- Flax is a neural network library for JAX that is designed for flexibility.☆7,082Updated this week
- Development repository for the Triton language and compiler☆18,460Feb 22, 2026Updated last week
- Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more☆34,940Updated this week
- Lightweight C++ and Python interface for datasets in zarr and N5 format☆130Jan 9, 2026Updated last month
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆549Updated this week
- A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆1,075Apr 17, 2024Updated last year
- WebGL-based viewer for volumetric data☆1,284Feb 21, 2026Updated last week
- Foundation Architecture for (M)LLMs☆3,135Apr 11, 2024Updated last year
- A composable and fully extensible C++ execution engine library for data management systems.☆4,065Updated this week
- JAX-based neural network library☆3,191Updated this week
- Cloud-friendly access to archival data☆360Jan 21, 2026Updated last month
- jax-triton contains integrations between JAX and OpenAI Triton☆439Updated this week
- Next-generation file format (NGFF) specifications for storing bioimaging data in the cloud.☆153Feb 16, 2026Updated last week
- Training and serving large-scale neural networks with auto parallelization.☆3,183Dec 9, 2023Updated 2 years ago
- Orbax provides common checkpointing and persistence utilities for JAX users☆482Updated this week
- common in-memory tensor structure☆1,169Jan 26, 2026Updated last month
- A machine learning compiler for GPUs, CPUs, and ML accelerators☆4,007Updated this week
- High performance model preprocessing library on PyTorch☆646Mar 29, 2024Updated last year
- Simple, safe way to store and distribute tensors☆3,637Updated this week
- ☆2,946Jan 15, 2026Updated last month
- A minimal Zarr image viewer based on Viv.☆184Feb 1, 2026Updated last month
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆1,027Dec 17, 2025Updated 2 months ago
- Scalable array processing with bounded memory☆256Feb 6, 2026Updated 3 weeks ago
- ☆135Dec 22, 2025Updated 2 months ago
- Pydantic models for OME-NGFF☆24Sep 14, 2025Updated 5 months ago
- Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.☆41,516Updated this week
- Distributed Xarray with Apache Beam☆165Jan 14, 2026Updated last month
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,401Feb 20, 2026Updated last week
- Performance-portable, length-agnostic SIMD with runtime dispatch☆5,346Updated this week
- cuDF - GPU DataFrame Library☆9,498Updated this week
- PyTorch extensions for high performance and large scale training.☆3,400Apr 26, 2025Updated 10 months ago
- Scalable Optical Flow-based Image Montaging and Alignment☆87Jan 26, 2026Updated last month
- A retargetable MLIR-based machine learning compiler and runtime toolkit.☆3,614Updated this week
- ☆65Oct 31, 2025Updated 4 months ago
- Task-based datasets, preprocessing, and evaluation for sequence models.☆594Feb 3, 2026Updated 3 weeks ago
- DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.☆41,648Updated this week