164 packages found. Page 1 of 4.

1 2 3 4 Next › Last »

Name Version Votes Popularity? Description Maintainer Last Updated
python-transformers 4.41.1-1 8 0.56 State-of-the-art Natural Language Processing for Jax, PyTorch and TensorFlow xiota 2024-05-23 08:42 (UTC)
r-torch 0.13.0-1 0 0.00 Tensors and Neural Networks with 'GPU' Acceleration BioArchLinuxBot 2024-05-22 00:07 (UTC)
python-timm 1.0.3-1 0 0.00 PyTorch Image Models hottea 2024-05-15 21:09 (UTC)
python-accelerate 0.30.1-1 1 0.56 A simple way to train and use PyTorch models with multi-GPU, TPU, mixed-precision daskol 2024-05-10 19:28 (UTC)
python-pytorch-msssim-git r155.b057b07-2 0 0.00 Fast and differentiable MS-SSIM and SSIM for pytorch trougnouf 2024-05-09 13:28 (UTC)
python-lion-pytorch 0.1.4-1 0 0.00 🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch Xangelix 2024-05-06 14:01 (UTC)
libtorch-cxx11abi-cuda 2.3.0-1 0 0.00 Tensors and Dynamic neural networks in Python with strong GPU acceleration (with CXX11 ABI) (with CUDA support) daizhirui 2024-05-06 09:29 (UTC)
libtorch-cxx11abi-rocm 2.3.0-1 0 0.00 Tensors and Dynamic neural networks in Python with strong GPU acceleration (with CXX11 ABI) (with ROCM support) daizhirui 2024-05-06 09:29 (UTC)
libtorch-cxx11abi-cpu 2.3.0-1 0 0.00 Tensors and Dynamic neural networks in Python with strong GPU acceleration (with CXX11 ABI) (CPU only) daizhirui 2024-05-06 09:29 (UTC)
python-pytorch-rocm-bin 2.3.0-2 0 0.00 Tensors and Dynamic neural networks in Python with strong GPU acceleration (binary release) trougnouf 2024-05-02 14:09 (UTC)
python-torchvision-rocm-bin 0.18.0-3 0 0.00 Datasets, transforms, and models specific to computer vision (binary release) trougnouf 2024-05-02 14:09 (UTC)
python-torchaudio 2.3.0-1 5 0.00 Data manipulation and transformation for audio signal processing, powered by PyTorch NBonaparte 2024-04-29 05:17 (UTC)
python-horovod 0.28.1-4 0 0.00 Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet hottea 2024-04-28 14:30 (UTC)
python-ttach 0.0.3-4 0 0.00 Image Test Time Augmentation with PyTorch hottea 2024-04-28 13:53 (UTC)
python-unet 0.7.9-3 0 0.00 PyTorch Implementation of 2D and 3D U-Net hottea 2024-04-28 11:36 (UTC)
python-einops 0.8.0-1 2 0.62 Deep learning operations reinvented (for pytorch, tensorflow, jax and others) hottea 2024-04-28 04:28 (UTC)
python-torchtext 1:0.18.0-1 3 0.00 Data loaders and abstractions for text and NLP hottea 2024-04-24 21:31 (UTC)
torchvision-rocm 0.17.2-1 5 0.97 Datasets, transforms, and models specific to computer vision (C++ library only, with ROCM support) wuxxin 2024-04-24 20:53 (UTC)
python-torchvision-rocm 0.17.2-1 5 0.97 Datasets, transforms, and models specific to computer vision (with ROCM support) wuxxin 2024-04-24 20:53 (UTC)
python-apex-git 22.03.r237.gb5df1ccf-1 0 0.00 A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch petronny 2024-04-24 13:02 (UTC)
libtorch-cuda 2.2.2-2 0 0.00 PyTorch C++ API chengscott 2024-04-08 10:03 (UTC)
python-pytorch-cxx11abi 2.2.2-1 0 0.00 Tensors and Dynamic neural networks in Python with strong GPU acceleration (with CXX11 ABI) daizhirui 2024-04-01 19:15 (UTC)
python-pytorch-cxx11abi-opt 2.2.2-1 0 0.00 Tensors and Dynamic neural networks in Python with strong GPU acceleration (with CXX11 ABI) (with AVX2 CPU optimizations) daizhirui 2024-04-01 19:15 (UTC)
python-pytorch-cxx11abi-cuda 2.2.2-1 0 0.00 Tensors and Dynamic neural networks in Python with strong GPU acceleration (with CXX11 ABI) (with CUDA) daizhirui 2024-04-01 19:15 (UTC)
python-pytorch-cxx11abi-opt-cuda 2.2.2-1 0 0.00 Tensors and Dynamic neural networks in Python with strong GPU acceleration (with CXX11 ABI) (with CUDA and AVX2 CPU optimizations) daizhirui 2024-04-01 19:15 (UTC)
python-pytorch-cxx11abi-rocm 2.2.2-1 0 0.00 Tensors and Dynamic neural networks in Python with strong GPU acceleration (with CXX11 ABI) (with ROCm) daizhirui 2024-04-01 19:15 (UTC)
python-pytorch-cxx11abi-opt-rocm 2.2.2-1 0 0.00 Tensors and Dynamic neural networks in Python with strong GPU acceleration (with CXX11 ABI) (with ROCm and AVX2 CPU optimizations) daizhirui 2024-04-01 19:15 (UTC)
libkineto-git r505.6968a24-1 0 0.00 Kineto is part of the PyTorch Profiler daizhirui 2024-03-31 06:36 (UTC)
torch-tb-profiler-git r505.6968a24-1 0 0.00 Kineto is part of the PyTorch Profiler daizhirui 2024-03-31 06:36 (UTC)
python-torchmetrics 1.3.2-1 2 0.56 Machine learning metrics for distributed, scalable PyTorch applications alhirzel 2024-03-31 02:56 (UTC)
python-pytorch-lightning 2.2.1-1 4 0.00 The lightweight PyTorch wrapper for high-performance AI research alhirzel 2024-03-31 02:55 (UTC)
python-visdom 0.2.4-5 0 0.00 A flexible tool for creating, organizing, and sharing visualizations of live, rich data. Supports Torch and Numpy. hottea 2024-03-30 06:37 (UTC)
python-pretrainedmodels-git r154.8aae3d8-5 0 0.00 Pretrained ConvNets for PyTorch hottea 2024-03-21 20:18 (UTC)
python-diffusers 0.27.2-1 3 0.00 Pretrained diffusion models for image and audio generation in PyTorch xiota 2024-03-20 12:18 (UTC)
python-pomegranate 1.0.4-1 0 0.00 A PyTorch implementation of probabilistic models carlosal1015 2024-03-15 02:23 (UTC)
python-thop 1:0.1.1.post2209072238-1 0 0.00 Count the MACs / FLOPs of your PyTorch model. fanyujun 2024-03-06 07:42 (UTC)
python-qualia-codegen-core 2.2.0-1 0 0.00 Convert a pre-trained Keras .h5 or PyTorch model to C code for inference. piernov 2024-03-01 13:43 (UTC)
python-pytorch3d 0.7.6-1 0 0.00 Provides efficient, reusable components for 3D Computer Vision research with PyTorch. piernov 2024-02-22 19:28 (UTC)
python-torchio 0.19.6-1 0 0.00 Tools for medical image processing in deep learning and PyTorch hottea 2024-02-20 05:22 (UTC)
python-torchattacks 3.5.1-1 1 0.12 PyTorch implementation of adversarial attacks Xeonacid 2024-02-11 12:47 (UTC)
python-tinygrad 0.8.0-1 0 0.00 You like pytorch? You like micrograd? You love tinygrad! <3 pravinxor 2024-02-06 16:16 (UTC)
python-torchsde 0.2.6-1 0 0.00 Differentiable SDE solvers with GPU support and efficient sensitivity analysis Henry-ZHR 2024-01-31 00:46 (UTC)
python-pytorch-tensorrt 1.4.0-1 0 0.00 PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT sl1pkn07 2024-01-25 19:54 (UTC)
python-open-clip-torch 2.24.0-1 1 0.00 Open source implementation of OpenAI's CLIP (Contrastive Language-Image Pre-training). alhirzel 2024-01-17 17:32 (UTC)
python-neurodiffeq 0.6.3-1 0 0.00 A light-weight & flexible library for solving differential equations using neural networks based on PyTorch carlosal1015 2024-01-04 17:20 (UTC)
python-torchinfo 1.8.0-1 0 0.00 Model summary in PyTorch, based off of the original torchsummary yodaembedding 2023-12-27 01:48 (UTC)
python-grad-cam 1.5.0-1 0 0.00 Image Test Time Augmentation with PyTorch hottea 2023-12-19 21:22 (UTC)
python-torch-geometric 2.4.0-1 0 0.00 Graph Neural Network Library for PyTorch acxz 2023-12-17 15:22 (UTC)
python-torch-scatter 2.1.2-1 0 0.00 PyTorch Extension Library of Optimized Scatter Operations acxz 2023-12-17 14:31 (UTC)
python-torch-sparse 0.6.18-1 0 0.00 PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations acxz 2023-12-17 13:57 (UTC)

164 packages found. Page 1 of 4.

1 2 3 4 Next › Last »