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: 2025-01-08 00:20:24 UTC
- JSON Representation
https://github.com/ExponentialML/Text-To-Video-Finetuning
Finetune ModelScope's Text To Video model using Diffusers 🧨
deep-learning diffusers diffusion-models modelscope pytorch stable-diffusion text-to-video text2video
Last synced: 10 Nov 2024
https://github.com/xwying/torchshow
Visualize PyTorch tensors with a single line of code.
image-tensor notebook pytorch tensor visualization
Last synced: 05 Jan 2025
https://github.com/swz30/MIRNet
[ECCV 2020] Learning Enriched Features for Real Image Restoration and Enhancement. SOTA results for image denoising, super-resolution, and image enhancement.
attention-mechanism computer-vision eccv2020 image-denoising image-enhancement image-restoration low-level-vision multi-resolution-streams pytorch super-resolution
Last synced: 03 Nov 2024
https://github.com/iMoonLab/DeepHypergraph
A pytorch library for graph and hypergraph computation.
deep-learning graph-neural-networks hypergraph-neural-networks pytorch
Last synced: 28 Oct 2024
https://github.com/wvangansbeke/lanedetection_end2end
End-to-end Lane Detection for Self-Driving Cars (ICCV 2019 Workshop)
computer-vision deep-learning end-to-end lane-detection least-squares pytorch self-driving-cars
Last synced: 04 Jan 2025
https://github.com/demiseom/specaugment
A Implementation of SpecAugment with Tensorflow & Pytorch, introduced by Google Brain
data-augmentation python pytorch specaugment speech speech-recognition tensorflow
Last synced: 10 Jan 2025
https://github.com/fastai/fastai2
Temporary home for fastai v2 while it's being developed
data-science deep-learning fastai jupyter machine-learning nbdev python pytorch
Last synced: 27 Nov 2024
https://github.com/ymcui/MacBERT
Revisiting Pre-trained Models for Chinese Natural Language Processing (MacBERT)
bert language-model macbert nlp pytorch tensorflow transformers
Last synced: 16 Nov 2024
https://github.com/pytorch/torcharrow
High performance model preprocessing library on PyTorch
Last synced: 07 Oct 2024
https://github.com/sgrvinod/a-pytorch-tutorial-to-super-resolution
Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network | a PyTorch Tutorial to Super-Resolution
gan gans generative-adversarial-network pytorch pytorch-tutorial srgan super-resolution
Last synced: 04 Jan 2025
https://github.com/project-monai/monailabel
MONAI Label is an intelligent open source image labeling and learning tool.
3d 3d-slicer-extension active-learning deep-learning machine-learning medical-imaging monai pytorch segmentation
Last synced: 10 Jan 2025
https://github.com/blackhc/tfpyth
Putting TensorFlow back in PyTorch, back in TensorFlow (differentiable TensorFlow PyTorch adapters).
machine-learning pytorch tensorflow
Last synced: 10 Jan 2025
https://github.com/maghoumi/pytorch-softdtw-cuda
Fast CUDA implementation of (differentiable) soft dynamic time warping for PyTorch
cuda deep-learning dynamic-time-warping pytorch soft-dtw
Last synced: 05 Jan 2025
https://github.com/wvangansbeke/LaneDetection_End2End
End-to-end Lane Detection for Self-Driving Cars (ICCV 2019 Workshop)
computer-vision deep-learning end-to-end lane-detection least-squares pytorch self-driving-cars
Last synced: 14 Nov 2024
https://github.com/Megvii-BaseDetection/cvpods
All-in-one Toolbox for Computer Vision Research.
3d classification computer-vision cvpods detection keypoints pytorch research segmentation self-supervised-learning
Last synced: 28 Oct 2024
https://github.com/lucasjinreal/dcnv2_latest
DCNv2 supports decent pytorch such as torch 1.5+ (now 1.8+)
Last synced: 04 Jan 2025
https://github.com/zkywsg/daily-deeplearning
🔥机器学习/深度学习/Python/大模型/多模态/LLM/deeplearning/Python/Algorithm interview/NLP Tutorial
cv deep-learning leetcode leetcode-python leetcode-solutions llm machine-learning nlp python pytorch pytorch-nlp pytorch-tutorial pytorch-tutorials tensorflow tensorflow-examples tensorflow-tutorials
Last synced: 08 Jan 2025
https://github.com/DemisEom/SpecAugment
A Implementation of SpecAugment with Tensorflow & Pytorch, introduced by Google Brain
data-augmentation python pytorch specaugment speech speech-recognition tensorflow
Last synced: 27 Nov 2024
https://github.com/BlackHC/tfpyth
Putting TensorFlow back in PyTorch, back in TensorFlow (differentiable TensorFlow PyTorch adapters).
machine-learning pytorch tensorflow
Last synced: 28 Oct 2024
https://github.com/noclin/lightmirrors
LightMirrors is a lightweight mirror server with caching capabilities that currently supports DockerHub, K8S, PyPI, PyTorch, and NPM.
cache-proxy docker docker-compose dockerhub ghcr k8s mirror mirror-site mirrors npm nvcr pypi python pytorch quay reverse-proxy
Last synced: 04 Jan 2025
https://github.com/wanmeihuali/taichi_3d_gaussian_splatting
An unofficial implementation of paper 3D Gaussian Splatting for Real-Time Radiance Field Rendering by taichi lang.
3d-reconstruction 3d-rendering computer-graphics computer-vision machine-learning nerf python pytorch real-time-rendering taichi
Last synced: 07 Nov 2024
https://github.com/yuliangxiu/mobilepose
Light-weight Single Person Pose Estimator
data-augmentation dataloader deep-learning deeppose dsntnn heatmap lightweight machine-learning mobile-device mobilenetv2 pose-estimation pytorch real-time realtime resnet-18 shufflenet shufflenet-v2 shufflenetv2 squeezenet
Last synced: 08 Jan 2025
https://github.com/blacksamorez/tensor_parallel
Automatically split your PyTorch models on multiple GPUs for training & inference
deep-learning machine-learning natural-language-processing nlp python pytorch pytorch-transformers
Last synced: 04 Jan 2025
https://github.com/lcswillems/rl-starter-files
RL starter files in order to immediately train, visualize and evaluate an agent without writing any line of code
a2c a3c minigrid multi-process ppo preprocessed-observations pytorch reward-shaping
Last synced: 27 Nov 2024
https://github.com/seungwonpark/melgan
MelGAN vocoder (compatible with NVIDIA/tacotron2)
gan neural-vocoder pytorch tts
Last synced: 04 Jan 2025
https://github.com/grok-ai/nn-template
Generic template to bootstrap your PyTorch project.
best-practices best-practises cookiecutter deep-learning dvc github-actions huggingface-datasets hydra mkdocs pre-commit project-structure pytorch pytorch-lightning reproducibility research streamlit template wandb weights-and-biases
Last synced: 10 Jan 2025
https://github.com/YuliangXiu/MobilePose
Light-weight Single Person Pose Estimator
data-augmentation dataloader deep-learning deeppose dsntnn heatmap lightweight machine-learning mobile-device mobilenetv2 pose-estimation pytorch real-time realtime resnet-18 shufflenet shufflenet-v2 shufflenetv2 squeezenet
Last synced: 27 Nov 2024
https://github.com/monologg/JointBERT
Pytorch implementation of JointBERT: "BERT for Joint Intent Classification and Slot Filling"
bert intent-classification joint-bert pytorch slot-filling slu transformers
Last synced: 06 Nov 2024
https://github.com/pytorchbearer/torchbearer
torchbearer: A model fitting library for PyTorch
deep-learning differentiable-programming machine-learning model-fitting python3 pytorch
Last synced: 07 Nov 2024
https://github.com/mit-han-lab/pvcnn
[NeurIPS 2019, Spotlight] Point-Voxel CNN for Efficient 3D Deep Learning
frustum-pointnet frustum-pointnets kitti point-cloud point-voxel-cnn pointnet pointnet2 pvcnn pytorch s3dis shapenet
Last synced: 28 Oct 2024
https://github.com/qdrant/quaterion
Blazing fast framework for fine-tuning similarity learning models
contrastive-learning cosine-similarity deep-learning knn machine-learning metric-learning nearest-neighbor-search python pytorch pytorch-lightning similarity-learning similarity-search
Last synced: 01 Nov 2024
https://github.com/patrick-kidger/neuralcde
Code for "Neural Controlled Differential Equations for Irregular Time Series" (Neurips 2020 Spotlight)
controlled-differential-equations deep-learning deep-neural-networks differential-equations dynamical-systems machine-learning neural-differential-equations neural-networks pytorch rough-paths time-series
Last synced: 04 Jan 2025
https://github.com/petarv-/dgi
Deep Graph Infomax (https://arxiv.org/abs/1809.10341)
deep-graph-infomax graph-convolutional-networks neural-networks python pytorch unsupervised-learning unsupervised-node-embedding
Last synced: 04 Jan 2025
https://github.com/DerrickXuNu/OpenCOOD
[ICRA 2022] An opensource framework for cooperative detection. Official implementation for OPV2V.
autonomous-driving collaborative-perception computer-vision deep-learning multi-agent-perception multi-agent-systems pytorch simulations
Last synced: 28 Oct 2024
https://github.com/SamsungLabs/ritm_interactive_segmentation
Reviving Iterative Training with Mask Guidance for Interactive Segmentation
hrnets interactive-segmentation pretrained-models pytorch segmentation
Last synced: 15 Nov 2024
https://github.com/yeyupiaoling/masr
Pytorch实现的流式与非流式的自动语音识别框架,同时兼容在线和离线识别,目前支持Conformer、Squeezeformer、DeepSpeech2模型,支持多种数据增强方法。
asr conformer deep-learning deepspeech pytorch speech speech-recognition speech-to-text squeezeformer
Last synced: 09 Jan 2025
https://github.com/songlab-cal/tape
Tasks Assessing Protein Embeddings (TAPE), a set of five biologically relevant semi-supervised learning tasks spread across different domains of protein biology.
benchmark dataset deep-learning language-modeling protein-sequences protein-structure pytorch semi-supervised-learning
Last synced: 07 Jan 2025
https://github.com/perone/euclidesdb
A multi-model machine learning feature embedding database
cpp database deep-learning machine-learning pytorch search
Last synced: 10 Jan 2025
https://github.com/YU1ut/MixMatch-pytorch
Code for "MixMatch - A Holistic Approach to Semi-Supervised Learning"
deep-learning pytorch semi-supervised-learning
Last synced: 15 Nov 2024
https://github.com/nianticlabs/manydepth
[CVPR 2021] Self-supervised depth estimation from short sequences
cityscapes cost-volumes cvpr cvpr2021 depth-estimation depths estimating-depths kitti monodepth pytorch self-supervised self-supervised-learning
Last synced: 04 Jan 2025
https://github.com/JackWoo0831/Yolov7-tracker
Yolo X, v7, v8 and several Multi-Object Tracker(SORT, DeepSORT, ByteTrack, BoT-SORT, etc.) in MOT17 and VisDrone2019 Dataset. It uses a unified style and integrated tracker for easy embedding in your own projects.
Last synced: 28 Oct 2024
https://github.com/zhihu/zhilight
A highly optimized LLM inference acceleration engine for Llama and its variants.
cpm cuda gpt inference-engine llama llm llm-serving minicpm pytorch qwen
Last synced: 10 Jan 2025
https://github.com/Ha0Tang/AttentionGAN
AttentionGAN for Unpaired Image-to-Image Translation & Multi-Domain Image-to-Image Translation
adversarial-networks attention-model computer-vision cyclegan deep-learning gans geometry ijcnn image-generation image-to-image-translation image-translation multi-domain pytorch unpaired
Last synced: 15 Nov 2024
https://github.com/BlackSamorez/tensor_parallel
Automatically split your PyTorch models on multiple GPUs for training & inference
deep-learning machine-learning natural-language-processing nlp python pytorch pytorch-transformers
Last synced: 29 Nov 2024
https://github.com/d-x-y/nas-bench-201
NAS-Bench-201 API and Instruction
automl dataset nas nas-bench-201 pytorch
Last synced: 08 Jan 2025
https://github.com/alibaba/Megatron-LLaMA
Best practice for training LLaMA models in Megatron-LM
deepspeed distributed-training llama llm megatron-lm pretraining pytorch
Last synced: 30 Oct 2024
https://github.com/araffin/robotics-rl-srl
S-RL Toolbox: Reinforcement Learning (RL) and State Representation Learning (SRL) for Robotics
arm baselines baxter-robot gym kuka pybullet pytorch reinforcement-learning representation-learning robotics state
Last synced: 06 Jan 2025
https://github.com/BobLiu20/YOLOv3_PyTorch
Full implementation of YOLOv3 in PyTorch
object-detection pytorch yolo yolov3
Last synced: 09 Nov 2024
https://github.com/yinboc/few-shot-meta-baseline
Meta-Baseline: Exploring Simple Meta-Learning for Few-Shot Learning, in ICCV 2021
few-shot-learning machine-learning pytorch
Last synced: 04 Jan 2025
https://github.com/maum-ai/faceshifter
Unofficial PyTorch Implementation for FaceShifter (https://arxiv.org/abs/1912.13457)
face-swapping pytorch pytorch-lightning
Last synced: 04 Jan 2025
https://github.com/morvanzhou/pytorch-a3c
Simple A3C implementation with pytorch + multiprocessing
a3c actor-critic asynchronous-advantage-actor-critic gym multiprocessing neural-network pytorch toy-example
Last synced: 11 Jan 2025
https://github.com/ternaus/robot-surgery-segmentation
Wining solution and its improvement for MICCAI 2017 Robotic Instrument Segmentation Sub-Challenge
computer-vision deep-learning image-segmentation medical-imaging python pytorch robot-assisted-surgery
Last synced: 06 Jan 2025
https://github.com/DanAnastasyev/DeepNLP-Course
Deep NLP Course
colab-notebook deep-learning keras nlp pytorch
Last synced: 19 Nov 2024
https://github.com/sunshineatnoon/Paper-Implementations
Use PyTorch to implement some classic frameworks
computer-vision deep-learning pytorch
Last synced: 27 Nov 2024
https://github.com/patrick-kidger/NeuralCDE
Code for "Neural Controlled Differential Equations for Irregular Time Series" (Neurips 2020 Spotlight)
controlled-differential-equations deep-learning deep-neural-networks differential-equations dynamical-systems machine-learning neural-differential-equations neural-networks pytorch rough-paths time-series
Last synced: 02 Nov 2024
https://github.com/ashishpatel26/real-time-ml-project
A curated list of applied machine learning and data science notebooks and libraries across different industries.
application deep-learning deeplearning dl keras machine-learning machine-learning-algorithms machinelearning ml ml-application project pytorch real-time real-time-data rl tensorflow theano
Last synced: 08 Jan 2025
https://github.com/lucastabelini/LaneATT
Code for the paper entitled "Keep your Eyes on the Lane: Real-time Attention-guided Lane Detection" (CVPR 2021)
computer-vision deep-learning lane-detection pytorch
Last synced: 28 Oct 2024
https://github.com/d2l-ai/d2l-vi
Một cuốn sách về Học Sâu đề cập đến nhiều framework phổ biến, được sử dụng trên 300 trường Đại học từ 55 đất nước bao gồm MIT, Stanford, Harvard, và Cambridge.
computer-vision d2l deep-learning kaggle keras machine-learning mlbvn mxnet python pytorch tensorflow vietnamese-language
Last synced: 04 Jan 2025
https://github.com/lucasjinreal/DCNv2_latest
DCNv2 supports decent pytorch such as torch 1.5+ (now 1.8+)
Last synced: 27 Oct 2024
https://github.com/chrischoy/FCGF
Fully Convolutional Geometric Features: Fast and accurate 3D features for registration and correspondence.
3d correspondence feature-extraction fully-convolutional minkowskiengine neural-network pytorch registration sparse-tensor
Last synced: 02 Nov 2024
https://github.com/mir-group/nequip
NequIP is a code for building E(3)-equivariant interatomic potentials
atomistic-simulations computational-chemistry deep-learning drug-discovery force-fields interatomic-potentials machine-learning materials-science molecular-dynamics pytorch
Last synced: 13 Nov 2024
https://github.com/cdqa-suite/cdQA
⛔ [NOT MAINTAINED] An End-To-End Closed Domain Question Answering System.
artificial-intelligence bert deep-learning information-retrieval natural-language-processing nlp pytorch question-answering reading-comprehension transformers
Last synced: 13 Nov 2024
https://github.com/sforaidl/kd_lib
A Pytorch Knowledge Distillation library for benchmarking and extending works in the domains of Knowledge Distillation, Pruning, and Quantization.
algorithm-implementations benchmarking data-science deep-learning-library knowledge-distillation machine-learning model-compression pruning pytorch quantization
Last synced: 05 Jan 2025
https://github.com/OndrejTexler/Few-Shot-Patch-Based-Training
The official implementation of our SIGGRAPH 2020 paper Interactive Video Stylization Using Few-Shot Patch-Based Training
deep-learning gan generative-adversarial-network image-generation image-to-image-translation pytorch style-transfer
Last synced: 15 Nov 2024
https://github.com/Lightning-Universe/lightning-transformers
Flexible components pairing 🤗 Transformers with :zap: Pytorch Lightning
hydra pytorch pytorch-lightning transformers
Last synced: 03 Nov 2024
https://github.com/alankbi/detecto
Build fully-functioning computer vision models with PyTorch
computer-vision faster-rcnn machine-learning object-detection python pytorch
Last synced: 25 Oct 2024
https://github.com/maudzung/ttnet-real-time-analysis-system-for-table-tennis-pytorch
Unofficial implementation of "TTNet: Real-time temporal and spatial video analysis of table tennis" (CVPR 2020)
dice-loss distributed-data-parallel event-detection event-spotting multi-task-learning pytorch segmentation spatial-video-analysis sport-analytics table-tennis tabletennis ttnet
Last synced: 08 Jan 2025
https://github.com/yanx27/everybodydancenow_reproduce_pytorch
Everybody dance now reproduced in pytorch
everybody-dance-now openpose pix2pixhd pose2pose pytorch
Last synced: 04 Jan 2025
https://github.com/augmentedstartups/AS-One
Easy & Modular Computer Vision Detectors, Trackers & SAM - Run YOLOv9,v8,v7,v6,v5,R,X in under 10 lines of code.
computer-vision deep-learning object-detection opencv pytorch sam tracking ultralytics yolor yolov5 yolov7 yolov8 yolov9 yolox
Last synced: 09 Nov 2024
https://github.com/sooftware/kospeech
Open-Source Toolkit for End-to-End Korean Automatic Speech Recognition leveraging PyTorch and Hydra.
asr attention-is-all-you-need conformer e2e-asr end-to-end jasper korean-speech ksponspeech las las-models pytorch seq2seq speech-recognition transformer
Last synced: 08 Jan 2025
https://github.com/alvinwan/neural-backed-decision-trees
Making decision trees competitive with neural networks on CIFAR10, CIFAR100, TinyImagenet200, Imagenet
cifar10 cifar100 decision-trees explainability image-classification imagenet interpretability neural-backed-decision-trees neural-networks pretrained-models pretrained-weights pytorch tiny-imagenet
Last synced: 05 Jan 2025
https://github.com/VincentStimper/normalizing-flows
PyTorch implementation of normalizing flow models
density-estimation glow invertible-neural-networks neural-spline-flow normalizing-flow pytorch real-nvp residual-flow variational-autoencoder variational-inference
Last synced: 17 Nov 2024
https://github.com/signatrix/efficientdet
(Pretrained weights provided) EfficientDet: Scalable and Efficient Object Detection implementation by Signatrix GmbH
coco-dataset deep-learning efficientdet efficientnet neural-network object-detection pytorch
Last synced: 27 Nov 2024
https://github.com/yanx27/EverybodyDanceNow_reproduce_pytorch
Everybody dance now reproduced in pytorch
everybody-dance-now openpose pix2pixhd pose2pose pytorch
Last synced: 04 Nov 2024
https://github.com/abeardear/pytorch-YOLO-v1
an experiment for yolo-v1, including training and testing.
experiment object-detection pytorch yolov1
Last synced: 09 Nov 2024
https://github.com/kwotsin/mimicry
[CVPR 2020 Workshop] A PyTorch GAN library that reproduces research results for popular GANs.
cgan cvpr cvpr20 cvpr2020 dcgan gan gans generative-adversarial-networks generative-models infomax-gan machine-learning python pytorch reproducibility sngan sngan-projection ssgan wgan-gp
Last synced: 08 Jan 2025
https://github.com/he-y/filter-pruning-geometric-median
Filter Pruning via Geometric Median for Deep Convolutional Neural Networks Acceleration (CVPR 2019 Oral)
model-compression pruning pytorch
Last synced: 04 Jan 2025
https://github.com/castorini/daam
Diffusion attentive attribution maps for interpreting Stable Diffusion.
diffusion explainable-ai generative-ai huggingface pytorch stable-diffusion
Last synced: 06 Jan 2025
https://github.com/freedomintelligence/textclassificationbenchmark
A Benchmark of Text Classification in PyTorch
attention-is-all-you-need benchmark capusle cnn cnn-classification crnn lstm lstm-sentiment-analysis pytorch quantum rcnn text-classification
Last synced: 08 Jan 2025
https://github.com/jiahuiyu/wdsr_ntire2018
Code of our winning entry to NTIRE super-resolution challenge, CVPR 2018
deep-neural-networks efficient-algorithm pytorch super-resolution wdsr
Last synced: 05 Jan 2025
https://github.com/webis-de/small-text
Active Learning for Text Classification in Python
active-learning deep-learning language-models looking-for-contributors machine-learning natural-language-processing nlp python pytorch small-language-models text-classification transformers
Last synced: 10 Jan 2025
https://github.com/kevinzakka/pytorch-goodies
PyTorch Boilerplate For Research
Last synced: 12 Nov 2024
https://github.com/JiahuiYu/wdsr_ntire2018
Code of our winning entry to NTIRE super-resolution challenge, CVPR 2018
deep-neural-networks efficient-algorithm pytorch super-resolution wdsr
Last synced: 26 Nov 2024
https://github.com/richzhang/colorization-pytorch
PyTorch reimplementation of Interactive Deep Colorization
computer-graphics computer-vision convolutional-neural-networks deep-learning image-colorization pytorch siggraph
Last synced: 05 Jan 2025
https://github.com/mkocabas/epipolarpose
Self-Supervised Learning of 3D Human Pose using Multi-view Geometry (CVPR2019)
3d-pose-estimation computer-vision cvpr cvpr-2019 cvpr19 cvpr2019 human-pose-estimation machine-learning pytorch self-supervised-learning
Last synced: 05 Jan 2025
https://github.com/mit-han-lab/lite-transformer
[ICLR 2020] Lite Transformer with Long-Short Range Attention
Last synced: 16 Nov 2024
https://github.com/qinzheng93/GeoTransformer
[CVPR2022] Geometric Transformer for Fast and Robust Point Cloud Registration
attention-mechanism computer-vision deep-learning neural-network point-cloud point-cloud-registration pytorch registration transformer
Last synced: 28 Oct 2024
https://github.com/jozu-ai/kitops
An open source DevOps tool for packaging and versioning AI/ML models, datasets, code, and configuration into an OCI artifact.
ai code datasets devops devops-tools gguf hacktoberfest kubernetes kubernetes-deployment ml mlops mlops-tools model-interpretability model-serving models opensource platform-engineering pytorch sklearn tensorflow
Last synced: 04 Jan 2025
https://github.com/aimagelab/mammoth
An Extendible (General) Continual Learning Framework based on Pytorch - official codebase of Dark Experience for General Continual Learning
continual-learning dark-experience-replay deep-learning der experience-replay knowledge-distillation neurips2020 pytorch
Last synced: 10 Jan 2025
https://github.com/bowenc0221/panoptic-deeplab
This is Pytorch re-implementation of our CVPR 2020 paper "Panoptic-DeepLab: A Simple, Strong, and Fast Baseline for Bottom-Up Panoptic Segmentation" (https://arxiv.org/abs/1911.10194)
bottom-up cityscapes cvpr2020 deeplab detectron2 instance-segmentation panoptic-segmentation pytorch semantic-segmentation sementation
Last synced: 04 Jan 2025
https://github.com/mikoto10032/automaticweightedloss
Multi-task learning using uncertainty to weigh losses for scene geometry and semantics, Auxiliary Tasks in Multi-task Learning
auxiliary-tasks deep-learning multi-task multi-task-learning pytorch weigh-losses
Last synced: 05 Jan 2025
https://github.com/pyg-team/pytorch-frame
Tabular Deep Learning Library for PyTorch
data-frame deep-learning pytorch tabular-learning
Last synced: 10 Jan 2025
https://github.com/Malllabiisc/CompGCN
ICLR 2020: Composition-Based Multi-Relational Graph Convolutional Networks
deep-learning graph-convolutional-networks graph-representation-learning iclr2020 link-prediction pytorch relation-embeddings
Last synced: 28 Nov 2024
https://github.com/neptune-ai/neptune-client
📘 The experiment tracker for foundation model training
comparison dl foundation keras learning lightgbm llm logger logging machine ml mlops monitoring optuna pytorch rl tensorflow versioning visualization xgboost
Last synced: 08 Jan 2025
https://github.com/castorini/hedwig
PyTorch deep learning models for document classification
deep-learning document-classification pytorch
Last synced: 06 Nov 2024
https://github.com/hirofumi0810/neural_sp
End-to-end ASR/LM implementation with PyTorch
asr attention attention-mechanism automatic-speech-recognition ctc language-model language-modeling pytorch rnn-transducer seq2seq sequence-to-sequence speech speech-recognition streaming transformer transformer-xl
Last synced: 12 Nov 2024
https://github.com/fxia22/stn.pytorch
pytorch version of spatial transformer networks
Last synced: 05 Jan 2025
https://github.com/AviSoori1x/makeMoE
From scratch implementation of a sparse mixture of experts language model inspired by Andrej Karpathy's makemore :)
deep-learning large-language-models llm mixture-of-experts neural-networks pytorch pytorch-implementation
Last synced: 05 Nov 2024
https://github.com/timbmg/sentence-vae
PyTorch Re-Implementation of "Generating Sentences from a Continuous Space" by Bowman et al 2015 https://arxiv.org/abs/1511.06349
deep-learning generative-model neural-network nlp ptb pytorch vae
Last synced: 10 Jan 2025