Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
PyTorch
PyTorch is an open source machine learning library based on the Torch library, used for applications such as computer vision and natural language processing, primarily developed by Facebook’s AI Research lab.
- GitHub: https://github.com/topics/pytorch
- Wikipedia: https://en.wikipedia.org/wiki/PyTorch
- Repo: https://github.com/pytorch/pytorch
- Created by: Facebook's AI Research lab (FAIR)
- Released: September 2016
- Related Topics: python, pytorch-tutorial,
- Last updated: 2024-11-10 00:23:23 UTC
- JSON Representation
https://github.com/yuliangxiu/icon
[CVPR'22] ICON: Implicit Clothed humans Obtained from Normals
3d-reconstruction animation avatar-generator cloth-simulation computer-graphics computer-vision human-pose-estimation implicit-functions mesh-deformation metaverse normal-maps pifu pifuhd pose-estimation pytorch smpl smpl-body smpl-model smplx virtual-humans
Last synced: 14 Oct 2024
https://github.com/dusty-nv/jetson-containers
Machine Learning Containers for NVIDIA Jetson and JetPack-L4T
containers docker dockerfiles jetson machine-learning numpy nvidia pandas pytorch ros-containers ros2-foxy scikit-learn tensorflow
Last synced: 13 Oct 2024
https://github.com/airaria/textbrewer
A PyTorch-based knowledge distillation toolkit for natural language processing
bert distillation knowledge nlp pytorch
Last synced: 09 Oct 2024
https://github.com/airaria/TextBrewer
A PyTorch-based knowledge distillation toolkit for natural language processing
bert distillation knowledge nlp pytorch
Last synced: 03 Nov 2024
https://github.com/kakaobrain/fast-autoaugment
Official Implementation of 'Fast AutoAugment' in PyTorch.
augmentation automated-machine-learning automl cnn computer-vision convolutional-neural-networks deep-learning distributed image-classification pytorch
Last synced: 29 Oct 2024
https://github.com/Microsoft/EdgeML
This repository provides code for machine learning algorithms for edge devices developed at Microsoft Research India.
bonsai classifier cpp deep-learning edge-computing edge-devices edge-machine-learning emi-rnn fastgrnn fastrnn iot-device machine-learning machine-learning-algorithms microsoft microsoft-research protonn pytorch resource-constrained-ml sensor tensorflow
Last synced: 13 Nov 2024
https://github.com/microsoft/EdgeML
This repository provides code for machine learning algorithms for edge devices developed at Microsoft Research India.
bonsai classifier cpp deep-learning edge-computing edge-devices edge-machine-learning emi-rnn fastgrnn fastrnn iot-device machine-learning machine-learning-algorithms microsoft microsoft-research protonn pytorch resource-constrained-ml sensor tensorflow
Last synced: 01 Nov 2024
https://github.com/ialhashim/densedepth
High Quality Monocular Depth Estimation via Transfer Learning
deep-learning deep-neural-networks depth-estimation encoder-decoder-model keras-tensorflow python pytorch transfer-learning
Last synced: 14 Oct 2024
https://github.com/microsoft/edgeml
This repository provides code for machine learning algorithms for edge devices developed at Microsoft Research India.
bonsai classifier cpp deep-learning edge-computing edge-devices edge-machine-learning emi-rnn fastgrnn fastrnn iot-device machine-learning machine-learning-algorithms microsoft microsoft-research protonn pytorch resource-constrained-ml sensor tensorflow
Last synced: 07 Oct 2024
https://github.com/chaofengc/iqa-pytorch
👁️ 🖼️ 🔥PyTorch Toolbox for Image Quality Assessment, including LPIPS, FID, NIQE, NRQM(Ma), MUSIQ, TOPIQ, NIMA, DBCNN, BRISQUE, PI and more...
blind-image-quality-assessment image-aesthetic-assessment image-quality-assessment iqa python pytorch pytorch-implementation
Last synced: 15 Oct 2024
https://github.com/rwightman/efficientdet-pytorch
A PyTorch impl of EfficientDet faithful to the original Google impl w/ ported weights
efficientdet efficientnet object-detection pytorch semantic-segmentation
Last synced: 21 Oct 2024
https://github.com/D-X-Y/AutoDL-Projects
Automated deep learning algorithms implemented in PyTorch.
autodl automl nas neural-architecture-search pytorch
Last synced: 09 Nov 2024
https://github.com/YuliangXiu/ICON
[CVPR'22] ICON: Implicit Clothed humans Obtained from Normals
3d-reconstruction animation avatar-generator cloth-simulation computer-graphics computer-vision human-pose-estimation implicit-functions mesh-deformation metaverse normal-maps pifu pifuhd pose-estimation pytorch smpl smpl-body smpl-model smplx virtual-humans
Last synced: 04 Aug 2024
https://github.com/ranahanocka/MeshCNN
Convolutional Neural Network for 3D meshes in PyTorch
3d computer-graphics convolutional-neural-networks geometric-deep-learning machine-learning mesh pytorch segmentation
Last synced: 28 Oct 2024
https://github.com/rwightman/gen-efficientnet-pytorch
Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS
caffe2 efficientnet fbnet mnasnet mobilenetv3 onnx pretrained-models pytorch
Last synced: 22 Oct 2024
https://github.com/kan-bayashi/ParallelWaveGAN
Unofficial Parallel WaveGAN (+ MelGAN & Multi-band MelGAN & HiFi-GAN & StyleMelGAN) with Pytorch
hifigan melgan neural-vocoder parallel-wavenet pytorch realtime speech-synthesis style-melgan text-to-speech tts vocoder wavenet
Last synced: 06 Nov 2024
https://github.com/tsurumeso/vocal-remover
Vocal Remover using Deep Neural Networks
audio deep-learning pytorch segmentation spectrogram vocal-separation
Last synced: 15 Oct 2024
https://github.com/d-x-y/autodl-projects
Automated deep learning algorithms implemented in PyTorch.
autodl automl nas neural-architecture-search pytorch
Last synced: 15 Oct 2024
https://github.com/kan-bayashi/parallelwavegan
Unofficial Parallel WaveGAN (+ MelGAN & Multi-band MelGAN & HiFi-GAN & StyleMelGAN) with Pytorch
hifigan melgan neural-vocoder parallel-wavenet pytorch realtime speech-synthesis style-melgan text-to-speech tts vocoder wavenet
Last synced: 11 Oct 2024
https://github.com/ELS-RD/kernl
Kernl lets you run PyTorch transformer models several times faster on GPU with a single line of code, and is designed to be easily hackable.
cuda cuda-kernel pytorch transformer triton
Last synced: 12 Nov 2024
https://github.com/ssbuild/chatglm_finetuning
chatglm 6b finetuning and alpaca finetuning
adalora chatglm deep-learning freeze ia3 lora p-tuning-v2 pytorch qlora sft
Last synced: 15 Oct 2024
https://github.com/els-rd/kernl
Kernl lets you run PyTorch transformer models several times faster on GPU with a single line of code, and is designed to be easily hackable.
cuda cuda-kernel pytorch transformer triton
Last synced: 14 Oct 2024
https://github.com/predibase/lorax
Multi-LoRA inference server that scales to 1000s of fine-tuned LLMs
fine-tuning gpt llama llm llm-inference llm-serving llmops lora model-serving pytorch transformers
Last synced: 10 Oct 2024
https://github.com/OpenPPL/ppq
PPL Quantization Tool (PPQ) is a powerful offline neural network quantization tool.
caffe cuda deep-learning neural-network onnx open-source pytorch quantization
Last synced: 28 Oct 2024
https://github.com/openppl-public/ppq
PPL Quantization Tool (PPQ) is a powerful offline neural network quantization tool.
caffe cuda deep-learning neural-network onnx open-source pytorch quantization
Last synced: 05 Oct 2024
https://github.com/openppl/ppq
PPL Quantization Tool (PPQ) is a powerful offline neural network quantization tool.
caffe cuda deep-learning neural-network onnx open-source pytorch quantization
Last synced: 15 Oct 2024
https://github.com/n00mkrad/flowframes
Flowframes Windows GUI for video interpolation using DAIN (NCNN) or RIFE (CUDA/NCNN)
dain gui ncnn pytorch rife video-interpolation
Last synced: 10 Nov 2024
https://github.com/BloodAxe/pytorch-toolbelt
PyTorch extensions for fast R&D prototyping and Kaggle farming
augmentation deep-learning focal-loss image-classification image-processing image-segmentation jaccard-loss kaggle machine-learning object-detection pipeline python pytorch segmentation test-time-augmentation tta
Last synced: 04 Nov 2024
https://github.com/gpleiss/efficient_densenet_pytorch
A memory-efficient implementation of DenseNets
deep-learning densenet pytorch
Last synced: 14 Oct 2024
https://github.com/lufficc/SSD
High quality, fast, modular reference implementation of SSD in PyTorch
computer-vision deep-learning object-detection pytorch ssd
Last synced: 28 Oct 2024
https://github.com/lufficc/ssd
High quality, fast, modular reference implementation of SSD in PyTorch
computer-vision deep-learning object-detection pytorch ssd
Last synced: 14 Oct 2024
https://github.com/devnag/pytorch-generative-adversarial-networks
A very simple generative adversarial network (GAN) in PyTorch
deep-learning gan gans machine-learning pytorch
Last synced: 26 Oct 2024
https://github.com/sniklaus/3d-ken-burns
an implementation of 3D Ken Burns Effect from a Single Image using PyTorch
cuda cupy deep-learning python pytorch
Last synced: 14 Oct 2024
https://github.com/sgrvinod/Deep-Tutorials-for-PyTorch
In-depth tutorials for implementing deep learning models on your own with PyTorch.
deep-learning-tutorial learning-pytorch neural-network-tutorials pytorch pytorch-tutorial pytorch-tutorials
Last synced: 12 Nov 2024
https://github.com/Eric-mingjie/rethinking-network-pruning
Rethinking the Value of Network Pruning (Pytorch) (ICLR 2019)
convolutional-neural-networks deep-learning network-pruning pytorch
Last synced: 12 Nov 2024
https://github.com/caogang/wgan-gp
A pytorch implementation of Paper "Improved Training of Wasserstein GANs"
Last synced: 14 Oct 2024
https://github.com/erikwijmans/pointnet2_pytorch
PyTorch implementation of Pointnet2/Pointnet++
hydra point-cloud pointnet2 pytorch pytorch-lightning
Last synced: 29 Oct 2024
https://github.com/huggingface/pytorch-openai-transformer-lm
🐥A PyTorch implementation of OpenAI's finetuned transformer language model with a script to import the weights pre-trained by OpenAI
language-model neural-networks openai pytorch transformer
Last synced: 29 Oct 2024
https://github.com/eric-mingjie/rethinking-network-pruning
Rethinking the Value of Network Pruning (Pytorch) (ICLR 2019)
convolutional-neural-networks deep-learning network-pruning pytorch
Last synced: 29 Oct 2024
https://github.com/google-research/big_transfer
Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.
convolutional-neural-networks deep-learning imagenet jax pytorch tensorflow2 transfer-learning
Last synced: 14 Oct 2024
https://github.com/tensorly/tensorly
TensorLy: Tensor Learning in Python.
cupy decomposition jax machine-learning mxnet numpy python pytorch regression tensor tensor-algebra tensor-decomposition tensor-factorization tensor-learning tensor-methods tensor-regression tensorflow tensorly
Last synced: 15 Oct 2024
https://github.com/BindsNET/bindsnet
Simulation of spiking neural networks (SNNs) using PyTorch.
dynamic gpu-computing machine-learning neurons pytorch reinforcement-learning simulation snn spiking-neural-networks stdp synapse
Last synced: 10 Nov 2024
https://github.com/sgrvinod/deep-tutorials-for-pytorch
In-depth tutorials for implementing deep learning models on your own with PyTorch.
deep-learning-tutorial learning-pytorch neural-network-tutorials pytorch pytorch-tutorial pytorch-tutorials
Last synced: 14 Oct 2024
https://github.com/google-research/torchsde
Differentiable SDE solvers with GPU support and efficient sensitivity analysis.
deep-learning deep-neural-networks differential-equations dynamical-systems neural-differential-equations pytorch stochastic-differential-equations stochastic-processes stochastic-volatility-models
Last synced: 15 Oct 2024
https://github.com/xinntao/edvr
Winning Solution in NTIRE19 Challenges on Video Restoration and Enhancement (CVPR19 Workshops) - Video Restoration with Enhanced Deformable Convolutional Networks. EDVR has been merged into BasicSR and this repo is a mirror of BasicSR.
Last synced: 29 Oct 2024
https://github.com/xinntao/EDVR
Winning Solution in NTIRE19 Challenges on Video Restoration and Enhancement (CVPR19 Workshops) - Video Restoration with Enhanced Deformable Convolutional Networks. EDVR has been merged into BasicSR and this repo is a mirror of BasicSR.
Last synced: 03 Nov 2024
https://github.com/IBM/pytorch-seq2seq
An open source framework for seq2seq models in PyTorch.
Last synced: 04 Nov 2024
https://github.com/nvidia/transformerengine
A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs, to provide better performance with lower memory utilization in both training and inference.
cuda deep-learning fp8 gpu jax machine-learning python pytorch
Last synced: 15 Oct 2024
https://github.com/junyanz/bicyclegan
Toward Multimodal Image-to-Image Translation
deep-learning gans generative-adversarial-network pix2pix pytorch
Last synced: 14 Oct 2024
https://github.com/bloodaxe/pytorch-toolbelt
PyTorch extensions for fast R&D prototyping and Kaggle farming
augmentation deep-learning focal-loss image-classification image-processing image-segmentation jaccard-loss kaggle machine-learning object-detection pipeline python pytorch segmentation test-time-augmentation tta
Last synced: 15 Oct 2024
https://github.com/tencent/turbotransformers
a fast and user-friendly runtime for transformer inference (Bert, Albert, GPT2, Decoders, etc) on CPU and GPU.
albert bert decoder gpt2 gpu huggingface-transformers inference machine-translation nlp pytorch roberta transformer
Last synced: 09 Oct 2024
https://github.com/Tencent/TurboTransformers
a fast and user-friendly runtime for transformer inference (Bert, Albert, GPT2, Decoders, etc) on CPU and GPU.
albert bert decoder gpt2 gpu huggingface-transformers inference machine-translation nlp pytorch roberta transformer
Last synced: 27 Oct 2024
https://github.com/open-mmlab/mmrazor
OpenMMLab Model Compression Toolbox and Benchmark.
autoslim classification darts detection knowledge-distillation nas pruning pytorch quantization segmentation spos
Last synced: 11 Nov 2024
https://github.com/junyanz/BicycleGAN
Toward Multimodal Image-to-Image Translation
deep-learning gans generative-adversarial-network pix2pix pytorch
Last synced: 03 Aug 2024
https://github.com/rusty1s/pytorch_scatter
PyTorch Extension Library of Optimized Scatter Operations
gather pytorch scatter segment
Last synced: 15 Oct 2024
https://github.com/erikwijmans/Pointnet2_PyTorch
PyTorch implementation of Pointnet2/Pointnet++
hydra point-cloud pointnet2 pytorch pytorch-lightning
Last synced: 28 Oct 2024
https://github.com/marcoslucianops/DeepStream-Yolo
NVIDIA DeepStream SDK 7.0 / 6.4 / 6.3 / 6.2 / 6.1.1 / 6.1 / 6.0.1 / 6.0 / 5.1 implementation for YOLO models
darknet deepstream nvidia nvidia-deepstream-sdk object-detection paddle paddlepaddle ppyoloe pytorch tensorrt ultralytics yolo yolor yolov3 yolov4 yolov5 yolov6 yolov7 yolov8 yolox
Last synced: 27 Oct 2024
https://github.com/marcoslucianops/deepstream-yolo
NVIDIA DeepStream SDK 7.0 / 6.4 / 6.3 / 6.2 / 6.1.1 / 6.1 / 6.0.1 / 6.0 / 5.1 implementation for YOLO models
darknet deepstream nvidia nvidia-deepstream-sdk object-detection paddle paddlepaddle ppyoloe pytorch tensorrt ultralytics yolo yolor yolov3 yolov4 yolov5 yolov6 yolov7 yolov8 yolox
Last synced: 15 Oct 2024
https://github.com/V2AI/Det3D
World's first general purpose 3D object detection codebse.
3d-object-detection autonomous-driving deep-learning kitti nuscenes object-detection point-cloud pytorch
Last synced: 28 Oct 2024
https://github.com/v2ai/det3d
World's first general purpose 3D object detection codebse.
3d-object-detection autonomous-driving deep-learning kitti nuscenes object-detection point-cloud pytorch
Last synced: 15 Oct 2024
https://github.com/poodarchu/Det3D
World's first general purpose 3D object detection codebse.
3d-object-detection autonomous-driving deep-learning kitti nuscenes object-detection point-cloud pytorch
Last synced: 13 Aug 2024
https://github.com/kubeflow/katib
Automated Machine Learning on Kubernetes
ai automl huggingface hyperparameter-tuning jax kubeflow kubernetes llm machine-learning mlops neural-architecture-search pytorch scikit-learn tensorflow
Last synced: 13 Oct 2024
https://github.com/nvidiagameworks/kaolin-wisp
NVIDIA Kaolin Wisp is a PyTorch library powered by NVIDIA Kaolin Core to work with neural fields (including NeRFs, NGLOD, instant-ngp and VQAD).
3d-deep-learning nerf neuralfields pytorch sdf
Last synced: 15 Oct 2024
https://github.com/kubeflow/training-operator
Distributed ML Training and Fine-Tuning on Kubernetes
ai distributed fine-tuning gpu huggingface jax kubeflow kubernetes llm machine-learning mlops python pytorch tensorflow xgboost
Last synced: 14 Oct 2024
https://github.com/kubeflow/tf-operator
Distributed ML Training and Fine-Tuning on Kubernetes
ai distributed fine-tuning gpu huggingface jax kubeflow kubernetes llm machine-learning mlops python pytorch tensorflow xgboost
Last synced: 19 Aug 2024
https://github.com/cszn/DnCNN
Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoising (TIP, 2017)
image-denoising jpeg-deblocking keras-tensorflow matconvnet pytorch residual-learning super-resolution
Last synced: 03 Nov 2024
https://github.com/emilianavt/OpenSeeFace
Robust realtime face and facial landmark tracking on CPU with Unity integration
cpu csharp depth-estimation detection-model face-detection face-landmarks face-tracking landmark-model mobilenetv3 onnx onnxruntime openseeface python pytorch tracker udp unity unity3d virtual-youtuber vtuber
Last synced: 03 Nov 2024
https://github.com/MLReef/mlreef
The collaboration workspace for Machine Learning
artificial-intelligence data-science deep-learning deeplearning machine-learning machine-learning-algorithms mlops mlops-environment models mxnet pytorch reproducibility tensorflow
Last synced: 27 Oct 2024
https://github.com/mlreef/mlreef
The collaboration workspace for Machine Learning
artificial-intelligence data-science deep-learning deeplearning machine-learning machine-learning-algorithms mlops mlops-environment models mxnet pytorch reproducibility tensorflow
Last synced: 29 Oct 2024
https://github.com/CoinCheung/BiSeNet
Add bisenetv2. My implementation of BiSeNet
ade20k bisenet cityscapes cocostuff ncnn openvino pytorch tensorrt triton-inference-server
Last synced: 31 Oct 2024
https://github.com/microsoft/NeuronBlocks
NLP DNN Toolkit - Building Your NLP DNN Models Like Playing Lego
artificial-intelligence deep-learning dnn knowledge-distillation model-compression natural-language-processing pytorch qna question-answering sequence-labeling text-classification text-matching
Last synced: 06 Nov 2024
https://github.com/toandaominh1997/efficientdet.pytorch
Implementation EfficientDet: Scalable and Efficient Object Detection in PyTorch
coco computer-vision demo detection efficientdet-d0 efficientnet focalloss multibox nms object-detection pascal-voc pytorch
Last synced: 29 Oct 2024
https://github.com/DeepGraphLearning/torchdrug
A powerful and flexible machine learning platform for drug discovery
deep-learning drug-discovery graph-neural-networks pytorch
Last synced: 12 Nov 2024
https://github.com/bindsnet/bindsnet
Simulation of spiking neural networks (SNNs) using PyTorch.
dynamic gpu-computing machine-learning neurons pytorch reinforcement-learning simulation snn spiking-neural-networks stdp synapse
Last synced: 09 Oct 2024
https://github.com/szagoruyko/attention-transfer
Improving Convolutional Networks via Attention Transfer (ICLR 2017)
attention deep-learning knowledge-distillation pytorch
Last synced: 29 Oct 2024
https://github.com/cszn/dncnn
Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoising (TIP, 2017)
image-denoising jpeg-deblocking keras-tensorflow matconvnet pytorch residual-learning super-resolution
Last synced: 29 Oct 2024
https://github.com/toandaominh1997/EfficientDet.Pytorch
Implementation EfficientDet: Scalable and Efficient Object Detection in PyTorch
coco computer-vision demo detection efficientdet-d0 efficientnet focalloss multibox nms object-detection pascal-voc pytorch
Last synced: 07 Aug 2024
https://github.com/NVIDIAGameWorks/kaolin-wisp
NVIDIA Kaolin Wisp is a PyTorch library powered by NVIDIA Kaolin Core to work with neural fields (including NeRFs, NGLOD, instant-ngp and VQAD).
3d-deep-learning nerf neuralfields pytorch sdf
Last synced: 03 Aug 2024
https://github.com/coincheung/bisenet
Add bisenetv2. My implementation of BiSeNet
ade20k bisenet cityscapes cocostuff ncnn openvino pytorch tensorrt triton-inference-server
Last synced: 14 Oct 2024
https://github.com/emilianavt/openseeface
Robust realtime face and facial landmark tracking on CPU with Unity integration
cpu csharp depth-estimation detection-model face-detection face-landmarks face-tracking landmark-model mobilenetv3 onnx onnxruntime openseeface python pytorch tracker udp unity unity3d virtual-youtuber vtuber
Last synced: 10 Oct 2024
https://github.com/keyu-tian/SparK
[ICLR'23 Spotlight🔥] The first successful BERT/MAE-style pretraining on any convolutional network; Pytorch impl. of "Designing BERT for Convolutional Networks: Sparse and Hierarchical Masked Modeling"
bert cnn convnet convolutional-neural-networks deep-learning iclr iclr2023 instance-segmentation mae mask-rcnn masked-autoencoder masked-image-modeling object-detection pre-trained-model pretrain pretraining pytorch self-supervised-learning sparse-convolution ssl
Last synced: 28 Oct 2024
https://github.com/deepgraphlearning/torchdrug
A powerful and flexible machine learning platform for drug discovery
deep-learning drug-discovery graph-neural-networks pytorch
Last synced: 15 Oct 2024
https://github.com/kumar-shridhar/PyTorch-BayesianCNN
Bayesian Convolutional Neural Network with Variational Inference based on Bayes by Backprop in PyTorch.
aleatoric-uncertainties bayes bayes-by-backprop bayesian-convnets bayesian-deep-learning bayesian-inference bayesian-network bayesian-networks bayesian-neural-networks bayesian-statistics convolutional-neural-networks image-recognition python pytorch pytorch-cnn variational-bayes variational-inference
Last synced: 30 Oct 2024
https://github.com/keyu-tian/spark
[ICLR'23 Spotlight🔥] The first successful BERT/MAE-style pretraining on any convolutional network; Pytorch impl. of "Designing BERT for Convolutional Networks: Sparse and Hierarchical Masked Modeling"
bert cnn convnet convolutional-neural-networks deep-learning iclr iclr2023 instance-segmentation mae mask-rcnn masked-autoencoder masked-image-modeling object-detection pre-trained-model pretrain pretraining pytorch self-supervised-learning sparse-convolution ssl
Last synced: 15 Oct 2024
https://github.com/speedinghzl/CCNet
CCNet: Criss-Cross Attention for Semantic Segmentation (TPAMI 2020 & ICCV 2019).
cityscape pytorch scene-parsing segmentation self-attention semantic-segmentation
Last synced: 31 Oct 2024
https://github.com/filipradenovic/cnnimageretrieval-pytorch
CNN Image Retrieval in PyTorch: Training and evaluating CNNs for Image Retrieval in PyTorch
cnn convolutional-neural-networks image-retrieval python pytorch
Last synced: 14 Oct 2024
https://github.com/speedinghzl/ccnet
CCNet: Criss-Cross Attention for Semantic Segmentation (TPAMI 2020 & ICCV 2019).
cityscape pytorch scene-parsing segmentation self-attention semantic-segmentation
Last synced: 29 Oct 2024
https://github.com/kumar-shridhar/pytorch-bayesiancnn
Bayesian Convolutional Neural Network with Variational Inference based on Bayes by Backprop in PyTorch.
aleatoric-uncertainties bayes bayes-by-backprop bayesian-convnets bayesian-deep-learning bayesian-inference bayesian-network bayesian-networks bayesian-neural-networks bayesian-statistics convolutional-neural-networks image-recognition python pytorch pytorch-cnn variational-bayes variational-inference
Last synced: 14 Oct 2024
https://github.com/vturrisi/solo-learn
solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning
barlow-twins byol contrastive-learning deepcluster dino mae masked-input-prediction moco nnclr nvidia-dali pytorch pytorch-lightning ressl self-supervised-learning simclr simsiam swav transformer-models vibcreg vicreg
Last synced: 15 Oct 2024
https://github.com/ydataai/ydata-synthetic
Synthetic data generators for tabular and time-series data
datageneration datagenerator deep-learning gan gan-architectures gans generative-adversarial-network machine-learning python3 pytorch synthetic-data tensorflow2 time-series timeseries training-data
Last synced: 15 Oct 2024
https://github.com/GOATmessi8/RFBNet
Receptive Field Block Net for Accurate and Fast Object Detection, ECCV 2018
detection mobilenet pytorch rfbnet
Last synced: 06 Nov 2024
https://github.com/nod-ai/shark-studio
SHARK Studio -- Web UI for SHARK+IREE High Performance Machine Learning Distribution
amd apple-silicon deep-learning machine-learning mlir nvidia pytorch
Last synced: 09 Oct 2024
https://github.com/deftruth/cuda-learn-notes
🎉 Modern CUDA Learn Notes with PyTorch: CUDA Cores, Tensor Cores, fp32/tf32, fp16/bf16, fp8/int8, flash_attn, rope, sgemm, hgemm, sgemv, warp/block reduce, elementwise, softmax, layernorm, rmsnorm.
block-reduce cuda cuda-programming elementwise flash-attention flash-attention-2 flash-attention-3 gemm gemv layernorm pytorch rmsnorm softmax triton warp-reduce
Last synced: 12 Nov 2024
https://github.com/goatmessi8/rfbnet
Receptive Field Block Net for Accurate and Fast Object Detection, ECCV 2018
detection mobilenet pytorch rfbnet
Last synced: 29 Oct 2024
https://github.com/breezedeus/pix2text
An Open-Source Python3 tool for recognizing layouts, tables, math formulas (LaTeX), and text in images, converting them into Markdown format. A free alternative to Mathpix, empowering seamless conversion of visual content into text-based representations. 80+ languages are supported.
image-to-markdown latex latex-pdf layout-analysis math-formula math-formula-recognition math-ocr mathpix ocr python pytorch table-ocr
Last synced: 11 Oct 2024