WeiTang114 / FMQ
A fast single-direction queue for multiprocessing.
☆33Updated 5 years ago
Related projects: ⓘ
- Multiprocessing queues for numpy arrays using shared memory☆54Updated last year
- PyProf2: PyTorch Profiling tool☆83Updated 4 years ago
- Serialize numpy arrays using msgpack☆194Updated 2 months ago
- Profile the GPU memory usage of every line in a Pytorch code☆82Updated 6 years ago
- Tool for reading and writing datasets of tensors in a Lightning Memory-Mapped Database (LMDB). Designed to manage machine learning datase…☆117Updated 4 years ago
- A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.☆39Updated 6 years ago
- A modular system for machinable research code☆34Updated last month
- Equi-normalization of Neural Networks☆115Updated 5 years ago
- Python API for writing multiprocessing pipelines☆85Updated 2 years ago
- MongoDB Browser for results created with the Sacred framework☆16Updated 5 years ago
- Configure Python functions explicitly and safely☆126Updated last year
- Python 3 Bindings for NVML library. Get NVIDIA GPU status inside your program.☆236Updated 2 years ago
- Python Binding to NVRTC☆79Updated 6 years ago
- Automatically insert nvtx ranges to PyTorch models☆17Updated 3 years ago
- Streamlined machine learning experiment management.☆106Updated 4 years ago
- Lightweight interface to AWS☆47Updated 4 years ago
- Example python (numpy) -- CUDA installable package with a C-extension library☆142Updated 5 years ago
- Simple implementation of the LSUV initialization in PyTorch☆58Updated 8 months ago
- ☆10Updated 4 years ago
- PyTorch implementation of the NIPS'17 paper Training Deep Networks without Learning Rates Through Coin Betting.☆38Updated 6 years ago
- Generate C++ and CUDA boilerplate for tensorflow custom operators☆20Updated 6 years ago
- ☆36Updated 5 years ago
- Implementation of Neural Arithmetic Logic Units (https://arxiv.org/pdf/1808.00508.pdf)☆31Updated 5 years ago
- A simple library for creating shared memory numpy arrays☆42Updated 3 years ago
- Utilities for Pytorch☆89Updated last year
- Small Python library to automatically set CUDA_VISIBLE_DEVICES to the least loaded device on multi-GPU systems.☆106Updated last year
- Pure Python dashboard for monitoring deep learning experiments (like TensorBoard for PyTorch/JAX/etc, without a browser)☆95Updated 2 years ago
- Odds and Ends and Things I've implemented.☆78Updated 5 years ago
- Implementation of "Learning with Random Learning Rates" in PyTorch.☆102Updated 4 years ago
- Python library for extracting mini-batches of data from a data source for the purpose of training neural networks☆86Updated 5 years ago