google / tensorstore

Related projects ⓘ

Alternatives and complementary repositories for tensorstore

• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆1,010Updated 7 months ago
• google / vizier
  Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.
  ☆1,482Updated this week
• pytorch / torcharrow
  High performance model preprocessing library on PyTorch
  ☆649Updated 7 months ago
• openxla / xla
  A machine learning compiler for GPUs, CPUs, and ML accelerators
  ☆2,712Updated 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 7 months ago
• rapidsai / raft
  RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-a…
  ☆780Updated this week
• TorchStudio / torchstudio
  IDE for PyTorch and its ecosystem
  ☆383Updated 6 months ago
• NVIDIA / aistore
  AIStore: scalable storage for AI applications
  ☆1,294Updated this week
• pytorch / functorch
  functorch is JAX-like composable function transforms for PyTorch.
  ☆1,395Updated this week
• pytorch / data
  A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.
  ☆1,134Updated this week
• dmlc / dlpack
  common in-memory tensor structure
  ☆911Updated last month
• dblalock / bolt
  10x faster matrix and vector operations
  ☆2,474Updated 2 years ago
• llvm / torch-mlir
  The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.
  ☆1,356Updated this week
• iree-org / iree
  A retargetable MLIR-based machine learning compiler and runtime toolkit.
  ☆2,851Updated this week
• rapidsai / rmm
  RAPIDS Memory Manager
  ☆492Updated this week
• aesara-devs / aesara
  Aesara is a Python library for defining, optimizing, and efficiently evaluating mathematical expressions involving multi-dimensional arra…
  ☆1,178Updated last 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,535Updated 9 months ago
• NVIDIA / MatX
  An efficient C++17 GPU numerical computing library with Python-like syntax
  ☆1,220Updated this week
• google / yggdrasil-decision-forests
  A library to train, evaluate, interpret, and productionize decision forest models such as Random Forest and Gradient Boosted Decision Tre…
  ☆498Updated this week
• data-apis / array-api
  RFC document, tooling and other content related to the array API standard
  ☆216Updated 3 weeks ago
• google-deepmind / alphatensor
  ☆2,689Updated 7 months ago
• google-deepmind / optax
  Optax is a gradient processing and optimization library for JAX.
  ☆1,705Updated last week
• hidet-org / hidet
  An open-source efficient deep learning framework/compiler, written in python.
  ☆652Updated last week
• NVIDIA-Merlin / dataloader
  The merlin dataloader lets you rapidly load tabular data for training deep leaning models with TensorFlow, PyTorch or JAX
  ☆408Updated 7 months ago
• google / fiddle
  ☆328Updated this week
• google-deepmind / dm-haiku
  JAX-based neural network library
  ☆2,909Updated 2 weeks ago
• google / autobound
  AutoBound automatically computes upper and lower bounds on functions.
  ☆355Updated 4 months ago
• google-deepmind / xmanager
  A platform for managing machine learning experiments
  ☆818Updated 3 months ago
• google-deepmind / dm_pix
  PIX is an image processing library in JAX, for JAX.
  ☆389Updated 2 weeks ago