Alternatives and similar repositories for torchmocks

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

• Lightning-AI / ecosystem-ci
  Automate issue discovery for your projects against Lightning nightly and releases.
  ☆46Updated 8 months ago
• lf1-io / padl
  Functional deep learning
  ☆108Updated 3 years ago
• teddykoker / learning-to-learn-jax
  JAX implementation of Learning to learn by gradient descent by gradient descent
  ☆28Updated 5 months ago
• speediedan / finetuning-scheduler
  A PyTorch Lightning extension that accelerates and enhances foundation model experimentation with flexible fine-tuning schedules.
  ☆67Updated this week
• Borda / pyDeprecate
  Smoothly deprecate and redirect Python functions/classes with smart warnings and auto-routing—keep your codebase clean while maintaining …
  ☆52Updated last week
• google-research / precondition
  ☆31Updated last week
• cgarciae / treex
  A Pytree Module system for Deep Learning in JAX
  ☆214Updated 2 years ago
• Quansight / pytest-pytorch
  pytest plugin for a better developer experience when working with the PyTorch test suite
  ☆44Updated 4 years ago
• rasbt / cyclemoid-pytorch
  Cyclemoid implementation for PyTorch
  ☆90Updated 3 years ago
• romesco / hydra-lightning
  Configuration classes enabling Hydra to configure and manage Pytorch Lightning projects.
  ☆43Updated 4 years ago
• MathInf / toroidal
  a lightweight transformer library for PyTorch
  ☆72Updated 4 years ago
• tchaton / pytorch2lightning
  ☆15Updated 4 years ago
• cgarciae / einop
  ☆60Updated 3 years ago
• iKernels / transformers-lightning
  A collection of Models, Datasets, DataModules, Callbacks, Metrics, Losses and Loggers to better integrate pytorch-lightning with transfor…
  ☆47Updated 2 years ago
• HomebrewML / TrueGrad
  PyTorch interface for TrueGrad Optimizers
  ☆43Updated 2 years ago
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆41Updated last year
• HomebrewML / HomebrewNLP-torch
  A case study of efficient training of large language models using commodity hardware.
  ☆68Updated 3 years ago
• SarthakYadav / audax
  A home for audio ML in JAX. Has common features, learnable frontends, pretrained supervised and self-supervised models.
  ☆69Updated 3 years ago
• meta-pytorch / torchfix
  TorchFix - a linter for PyTorch-using code with autofix support
  ☆152Updated 4 months ago
• fidelity / stoke
  A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…
  ☆67Updated 2 years ago
• pytorch / maskedtensor
  MaskedTensors for PyTorch
  ☆38Updated 3 years ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆47Updated last year
• LukasHedegaard / ride
  Training wheels, side rails, and helicopter parent for your Deep Learning projects in PyTorch
  ☆24Updated 2 years ago
• SkafteNicki / pl_crossvalidate
  ☆64Updated 2 years ago
• TomFrederik / grokking
  Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'
  ☆39Updated 4 years ago
• Lightning-Universe / paper-AAVAE
  ☆47Updated 3 years ago
• n2cholas / shapecheck
  Framework-agnostic library for checking array/tensor shapes at runtime.
  ☆46Updated 4 years ago
• ExpectationMax / sklearn-jax-kernels
  Composable kernels for scikit-learn implemented in JAX.
  ☆45Updated 5 years ago
• lucidrains / memory-editable-transformer
  My explorations into editing the knowledge and memories of an attention network
  ☆35Updated 3 years ago