Quansight / pytest-pytorch
pytest plugin for a better developer experience when working with the PyTorch test suite
☆44Updated 3 years ago
Alternatives and similar repositories for pytest-pytorch:
Users that are interested in pytest-pytorch are comparing it to the libraries listed below
- A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…☆67Updated last year
- Pytorch Lightning seed project with hydra☆18Updated 4 years ago
- Test pytorch code with minimal computational overhead☆25Updated last year
- An Easy To Use PyTorch Computer Vision Library☆51Updated last year
- Simple tooling for marking deprecated functions or classes and re-routing to the new successors' instance.☆51Updated 2 months ago
- a lightweight transformer library for PyTorch☆71Updated 3 years ago
- Large dataset storage format for Pytorch☆45Updated 3 years ago
- ☆15Updated 3 years ago
- A criticism of a recent paper on buggy image downsampling methods in popular image processing and deep learning libraries.☆72Updated 2 years ago
- (unofficial) - customized fork of DETR, optimized for intelligent obj detection on 'real world' custom datasets☆12Updated 4 years ago
- JAX implementation of Learning to learn by gradient descent by gradient descent☆27Updated 4 months ago
- An open source implementation of CLIP.☆32Updated 2 years ago
- A collection of Models, Datasets, DataModules, Callbacks, Metrics, Losses and Loggers to better integrate pytorch-lightning with transfor…☆47Updated last year
- ☆25Updated 3 years ago
- A collection of optimizers, some arcane others well known, for Flax.☆29Updated 3 years ago
- A very naive and simple benchmark between dlib and pytorch in terms of space and time☆19Updated 4 years ago
- Yet Another Neural Network Library 🤔☆25Updated last month
- GAN models implemented with Pytorch Lightning and Hydra configuration☆34Updated 2 years ago
- Image data augmentation scheduler for albumentations transforms☆19Updated 3 years ago
- Parameterized fit and prediction harnesses for pytorch☆39Updated 4 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- Texture mapping with variational auto-encoders☆40Updated 3 years ago
- EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax☆127Updated last year
- A lightweight library for tensorflow 2.0☆66Updated 5 years ago
- Hacks for PyTorch☆18Updated last year
- This repo parallelizes mAP_evaluation using python's multiprocessing module.☆18Updated 2 years ago
- ☆38Updated 2 years ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆85Updated last year
- Keras-like APIs for JAX framework☆50Updated last year
- Productionize machine learning predictions, with ONNX or without☆65Updated last year