• PyTorch

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

  • Referenced in 22 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 22 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 10 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...
  • AllenNLP

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

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

  • Referenced in 4 articles [sw30739]
  • Transformers (formerly known as pytorch-transformers and pytorch-pretrained-bert) provides state ... deep interoperability between TensorFlow 2.0 and PyTorch...
  • RLlib

  • Referenced in 6 articles [sw31153]
  • RLlib natively supports TensorFlow, TensorFlow Eager, and PyTorch, but most of its internals are framework...
  • 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...
  • TensorLy

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

  • Referenced in 3 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 3 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...
  • PyTorch-BigGraph

  • Referenced in 2 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...
  • AutoGraph

  • Referenced in 3 articles [sw30957]
  • learning, imperative style libraries like Autograd and PyTorch are easy to write, but suffer from...
  • TorchIO

  • Referenced in 3 articles [sw32330]
  • deep learning. It follows the design of PyTorch and relies on standard medical image processing...
  • medicaltorch

  • Referenced in 3 articles [sw32336]
  • MedicalTorch is an open-source framework for PyTorch, implementing an extensive set of loaders...
  • GPyTorch

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

  • Referenced in 1 article [sw35383]
  • Pkwrap: a PyTorch Package for LF-MMI Training of Acoustic Models. We present a simple ... useful to train acoustic models in PyTorch using Kaldi’s LF-MMI training framework ... wrapper, called pkwrap (short form of PyTorch kaldi wrapper), enables the user to utilize ... flexibility provided by PyTorch in designing model architectures. It exposes the LF-MMI cost function...
  • rTorch

  • Referenced in 1 article [sw33942]
  • package rTorch: R Bindings to ’PyTorch’. ’R’ implementation and interface of the Machine Learning platform ... PyTorch developed in ’Python’. It requires a ’conda’ environment with ’torch’ and ’torchvision ... provide ’PyTorch’ functions, methods and classes. The key object in ’PyTorch’ is the tensor which...