Alternatives and similar repositories for torchmocks:

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

• Quansight / pytest-pytorch
  pytest plugin for a better developer experience when working with the PyTorch test suite
  ☆44Updated 3 years ago
• tchaton / pytorch2lightning
  ☆15Updated 3 years ago
• Borda / pyDeprecate
  Simple tooling for marking deprecated functions or classes and re-routing to the new successors' instance.
  ☆51Updated 2 weeks ago
• rolandgvc / flaxvision
  A selection of neural network models ported from torchvision for JAX & Flax.
  ☆44Updated 4 years ago
• teddykoker / learning-to-learn-jax
  JAX implementation of Learning to learn by gradient descent by gradient descent
  ☆27Updated 6 months ago
• ColinQiyangLi / AdaCat
  AdaCat
  ☆49Updated 2 years ago
• Lightning-AI / ecosystem-ci
  Automate issue discovery for your projects against Lightning nightly and releases.
  ☆47Updated last week
• ganguli-lab / degrees-of-freedom
  ☆37Updated 3 years ago
• google-research / precondition
  ☆31Updated last week
• iKernels / transformers-lightning
  A collection of Models, Datasets, DataModules, Callbacks, Metrics, Losses and Loggers to better integrate pytorch-lightning with transfor…
  ☆47Updated last year
• fidelity / stoke
  A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…
  ☆67Updated last year
• f-dangel / sirfshampoo
  [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)
  ☆14Updated 5 months ago
• TomFrederik / grokking
  Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'
  ☆38Updated 3 years ago
• HomebrewML / TrueGrad
  PyTorch interface for TrueGrad Optimizers
  ☆41Updated last year
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆39Updated last year
• NeilGirdhar / tjax
  Tools for JAX
  ☆45Updated last week
• crowsonkb / dice-mc
  DiCE: The Infinitely Differentiable Monte-Carlo Estimator
  ☆31Updated last year
• nestordemeure / jochastic
  A JAX implementation of stochastic addition.
  ☆14Updated 2 years ago
• lucidrains / local-attention-flax
  Local Attention - Flax module for Jax
  ☆20Updated 3 years ago
• pytorch / maskedtensor
  MaskedTensors for PyTorch
  ☆38Updated 2 years ago
• f-dangel / vivit
  [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…
  ☆17Updated last year
• srush / g9py
  ☆19Updated last year
• SarthakYadav / audax
  A home for audio ML in JAX. Has common features, learnable frontends, pretrained supervised and self-supervised models.
  ☆68Updated 2 years ago
• michalwols / yann
  Yet Another Neural Network Library 🤔
  ☆27Updated 3 weeks ago
• tchaton / lightning-hydra-seed
  Pytorch Lightning seed project with hydra
  ☆18Updated 4 years ago
• cgarciae / einop
  ☆60Updated 3 years ago
• tmbdev-archive / pytorch-imagenet-wds
  ☆25Updated 4 years ago
• MathInf / toroidal
  a lightweight transformer library for PyTorch
  ☆71Updated 3 years ago
• unixpickle / vae-textures
  Texture mapping with variational auto-encoders
  ☆40Updated 3 years ago
• glouppe / flowing-with-jax
  ☆15Updated 4 years ago