• PyTorch

  • Referenced in 435 articles [sw20939]
  • PyTorch python package: Tensors and Dynamic neural networks in Python with strong GPU acceleration. PyTorch...
  • CycleGAN

  • Referenced in 66 articles [sw28302]
  • CycleGAN and pix2pix in PyTorch. We provide PyTorch implementations for both unpaired and paired image ... Park, and supported by Tongzhou Wang. This PyTorch implementation produces results comparable to or better...
  • pix2pix

  • Referenced in 65 articles [sw28303]
  • CycleGAN and pix2pix in PyTorch. We provide PyTorch implementations for both unpaired and paired image ... Park, and supported by Tongzhou Wang. This PyTorch implementation produces results comparable to or better...
  • torchdiffeq

  • Referenced in 59 articles [sw35082]
  • torchdiffeq: PyTorch Implementation of Differentiable ODE Solvers. This library provides ordinary differential equation (ODE) solvers ... implemented in PyTorch. Backpropagation through all solvers is supported using the adjoint method. For usage ... solvers are implemented in PyTorch, algorithms in this repository are fully supported...
  • Transformers

  • Referenced in 22 articles [sw30739]
  • Huggingface’s transformers: Transformers (formerly known as pytorch-transformers and pytorch-pretrained-bert) provides state ... deep interoperability between TensorFlow 2.0 and PyTorch...
  • GPyTorch

  • Referenced in 15 articles [sw35483]
  • scalable GP inference via BBMM, built on PyTorch...
  • AllenNLP

  • Referenced in 14 articles [sw26526]
  • open-source NLP research library, built on PyTorch. AllenNLP makes it easy to design...
  • Pyro

  • Referenced in 14 articles [sw27079]
  • written in Python and supported by PyTorch on the backend. Pyro enables flexible and expressive...
  • PyTorch Lightning

  • Referenced in 8 articles [sw35081]
  • PyTorch Lightning: The lightweight PyTorch wrapper for high-performance AI research. Scale your models...
  • TensorLy

  • Referenced in 12 articles [sw27020]
  • seamlessly perform computation with NumPy, MXNet, PyTorch, TensorFlow or CuPy, and run methods at scale...
  • RLlib

  • Referenced in 9 articles [sw31153]
  • RLlib natively supports TensorFlow, TensorFlow Eager, and PyTorch, but most of its internals are framework...
  • PyTorch-Kaldi

  • Referenced in 4 articles [sw38119]
  • PyTorch-Kaldi Speech Recognition Toolkit. The availability of open-source software is playing a remarkable ... develop state-of-the-art speech recognizers. PyTorch is used to build neural networks with ... thanks to its simplicity and flexibility. The PyTorch-Kaldi project aims to bridge ... efficiency of Kaldi and the flexibility of PyTorch. PyTorch-Kaldi is not only a simple...
  • PyTorch-BigGraph

  • Referenced in 4 articles [sw34086]
  • PyTorch-BigGraph: A Large-scale Graph Embedding System. Graph embedding methods produce unsupervised node features ... capability of existing embedding systems. We present PyTorch-BigGraph (PBG), an embedding system that incorporates...
  • ESPnet

  • Referenced in 6 articles [sw32059]
  • used dynamic neural network toolkits, Chainer and PyTorch, as a main deep learning engine. ESPnet...
  • BoTorch

  • Referenced in 6 articles [sw40383]
  • optimization of probabilistic models written in PyTorch, simplifying implementation of new acquisition functions. Our approach...
  • Kornia

  • Referenced in 4 articles [sw31197]
  • Open Source Differentiable Computer Vision Library for PyTorch. This work presents Kornia -- an open source ... generic computer vision problems. The package uses PyTorch as its main backend both for efficiency...
  • Captum

  • Referenced in 4 articles [sw35075]
  • unified and generic model interpretability library for PyTorch. In this paper we introduce a novel ... unified, open-source model interpretability library for PyTorch [12]. The library contains generic implementations...
  • PyG

  • Referenced in 4 articles [sw41050]
  • PyTorch Geometric) is a library built upon PyTorch to easily write and train Graph Neural...
  • Torchmeta

  • Referenced in 3 articles [sw30740]
  • Torchmeta: A Meta-Learning library for PyTorch. The constant introduction of standardized benchmarks ... Torchmeta, a library built on top of PyTorch that enables seamless and consistent evaluation ... abstraction. It also features some extensions for PyTorch to simplify the development of models compatible...