Alternatives and similar repositories for torchmocks

Users that are interested in torchmocks are comparing it to the libraries listed below

• tchaton / pytorch2lightning
  ☆15Updated 3 years ago
• Quansight / pytest-pytorch
  pytest plugin for a better developer experience when working with the PyTorch test suite
  ☆44Updated 3 years ago
• rolandgvc / flaxvision
  A selection of neural network models ported from torchvision for JAX & Flax.
  ☆44Updated 4 years ago
• Borda / pyDeprecate
  Simple tooling for marking deprecated functions or classes and re-routing to the new successors' instance.
  ☆51Updated last month
• teddykoker / learning-to-learn-jax
  JAX implementation of Learning to learn by gradient descent by gradient descent
  ☆27Updated 7 months ago
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆44Updated 11 months ago
• Lightning-AI / ecosystem-ci
  Automate issue discovery for your projects against Lightning nightly and releases.
  ☆47Updated last week
• unixpickle / vae-textures
  Texture mapping with variational auto-encoders
  ☆40Updated 3 years ago
• ColinQiyangLi / AdaCat
  AdaCat
  ☆49Updated 2 years ago
• tchaton / lightning-hydra-seed
  Pytorch Lightning seed project with hydra
  ☆18Updated 4 years ago
• iKernels / transformers-lightning
  A collection of Models, Datasets, DataModules, Callbacks, Metrics, Losses and Loggers to better integrate pytorch-lightning with transfor…
  ☆47Updated last year
• google-research / precondition
  ☆31Updated last month
• nestordemeure / jochastic
  A JAX implementation of stochastic addition.
  ☆14Updated 2 years ago
• fidelity / stoke
  A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…
  ☆67Updated last year
• cgarciae / einop
  ☆60Updated 3 years ago
• MathInf / toroidal
  a lightweight transformer library for PyTorch
  ☆71Updated 3 years ago
• cgarciae / jax_metrics
  A metrics library for the JAX ecosystem
  ☆40Updated 2 years ago
• crowsonkb / dice-mc
  DiCE: The Infinitely Differentiable Monte-Carlo Estimator
  ☆31Updated last year
• EleutherAI / tqdm-multiprocess
  Using queues, tqdm-multiprocess supports multiple worker processes, each with multiple tqdm progress bars, displaying them cleanly throug…
  ☆42Updated 4 years ago
• srush / g9py
  ☆19Updated last year
• speediedan / finetuning-scheduler
  A PyTorch Lightning extension that accelerates and enhances foundation model experimentation with flexible fine-tuning schedules.
  ☆63Updated last month
• lucidrains / einops-exts
  Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️
  ☆54Updated 2 years ago
• juliuskunze / cwvae-jax
  Clockwork VAEs in JAX/Flax
  ☆32Updated 3 years ago
• xflash96 / pybind11_package_example
  Example python package with pybind11 cpp extension
  ☆57Updated 4 years ago
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆39Updated last year
• google-deepmind / fancyflags
  A Python library for defining flat or nested dictionary flags.
  ☆31Updated last year
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• n2cholas / shapecheck
  Framework-agnostic library for checking array/tensor shapes at runtime.
  ☆46Updated 4 years ago
• fsschneider / cockpit-experiments
  Experiments for the NeurIPS 2021 paper "Cockpit: A Practical Debugging Tool for the Training of Deep Neural Networks"
  ☆13Updated 3 years ago
• LukasHedegaard / ride
  Training wheels, side rails, and helicopter parent for your Deep Learning projects in PyTorch
  ☆25Updated last year