Back to ag14774/diffdist
Open-source alternatives to Diffdist
30 open-source projects similar to ag14774/diffdist, ranked by how many features they have in common. Compare stars, activity and what each one does to find the best Diffdist alternative.
- lyken17/pytorch-opcounter
Lyken17/pytorch-OpCounter
5,080View on GitHubpytorch-OpCounter is a profiling utility for PyTorch neural networks designed to quantify model efficiency by calculating floating point operations and multiply-accumulate counts. It functions as a complexity analyzer to measure the computational cost and theoretical workload of different model architectures. The tool allows for the definition of custom operation counting rules to support third-party modules not covered by default. It uses forward hooks to intercept module calls and recursive traversal of the module tree to aggregate operations across child sub-modules. The project provides
Python - kevinmusgrave/pytorch-metric-learning
KevinMusgrave/pytorch-metric-learning
6,328View on GitHubPyTorch Metric Learning is an open-source library for training neural networks to produce similarity-preserving embedding spaces. It provides a modular framework where interchangeable loss functions, mining strategies, and evaluation tools can be composed to learn representations that map similar items to nearby points and dissimilar items to distant points in the embedding space. The library distinguishes itself through a highly configurable architecture that separates concerns across several interchangeable components. Users can assemble custom loss functions from pluggable distance metrics
Pythoncomputer-visioncontrastive-learningdeep-learning - erotemic/netharn
Erotemic/netharn
