Named tensors with first-class dimensions for PyTorch
☆334Jun 14, 2023Updated 3 years ago
Alternatives and similar repositories for torchdim
Users that are interested in torchdim are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- functorch is JAX-like composable function transforms for PyTorch.☆1,434Aug 21, 2025Updated last year
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,485May 2, 2025Updated last year
- A Pytree Module system for Deep Learning in JAX☆210Feb 26, 2023Updated 3 years ago
- ☆59Mar 8, 2022Updated 4 years ago
- TensorDict is a pytorch dedicated tensor container.☆1,043Updated this week
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/☆1,879Updated this week
- MaskedTensors for PyTorch☆38Jul 17, 2022Updated 4 years ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,627Aug 26, 2026Updated 2 months ago
- Memory mapped numpy arrays of varying shapes☆309Jun 24, 2026Updated 4 months ago
- A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆1,078Apr 17, 2024Updated 2 years ago
- A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.☆1,264Sep 16, 2026Updated last month
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,997Updated this week
- Tensors, for human consumption☆1,399Apr 9, 2026Updated 6 months ago
- A JAX implementation of stochastic addition.☆13Aug 15, 2022Updated 4 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Torch Distributed Experimental☆117Aug 5, 2024Updated 2 years ago
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆165Aug 19, 2026Updated 2 months ago
- Use Jax functions in Pytorch☆263Jul 1, 2023Updated 3 years ago
- Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch☆252Sep 1, 2022Updated 4 years ago
- KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows☆1,194Jul 17, 2026Updated 3 months ago
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️☆56Jan 5, 2023Updated 3 years ago
- Universal Notation for Tensor Operations in Python☆533Sep 12, 2026Updated last month
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆221Aug 24, 2026Updated 2 months ago
- ☆15Sep 24, 2022Updated 4 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Optimized library for large-scale extraction of frames and audio from video.☆205Sep 11, 2023Updated 3 years ago
- A concise but complete full-attention transformer with a set of promising experimental features from various papers☆5,961Oct 23, 2026Updated last week
- ☆958Oct 7, 2026Updated 3 weeks ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46May 4, 2021Updated 5 years ago
- High performance model preprocessing library on PyTorch☆639Mar 29, 2024Updated 2 years ago
- Fast Block Sparse Matrices for Pytorch☆551Jan 21, 2021Updated 5 years ago
- A High Level API for Deep Learning in JAX☆470Dec 15, 2022Updated 3 years ago
- Fast, differentiable sorting and ranking in PyTorch☆865Mar 3, 2026Updated 7 months ago
- A library to inspect and extract intermediate layers of PyTorch models.☆473May 12, 2022Updated 4 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- PIX is an image processing library in JAX, for JAX.☆452Updated this week
- ☆816Oct 20, 2026Updated last week
- Implementation of "compositional attention" from MILA, a multi-head attention variant that is reframed as a two-step attention process wi…☆51May 10, 2022Updated 4 years ago
- Optax is a gradient processing and optimization library for JAX.☆2,359Updated this week
- AI on the way. An auto deep learning pipe dream. An RDBMS approach to deep learning. Declarative, explainable, scalable, optimizable, eas…☆1,823Updated this week
- Multidimensional indexing for tensors☆143Jul 17, 2023Updated 3 years ago
- ☆31Oct 3, 2022Updated 4 years ago