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-07-29 13:55:27 UTC
- JSON Representation
https://github.com/csyanbin/TPN-pytorch
Pytorch Code for ICLR19 paper: Learning to Propagate Labels: Transductive Propagation Network for Few-shot Learning.
few-shot-learning label-propagation meta-learning pytorch transductive-few-shot-learning
Last synced: 30 Jul 2024
https://github.com/d-li14/PSConv
[ECCV 2020] PSConv: Squeezing Feature Pyramid into One Compact Poly-Scale Convolutional Layer
convolution eccv2020 feature-pyramids instance-segmentation mmdetection multi-scale object-detection pytorch
Last synced: 03 Aug 2024
https://github.com/pqhieu/jsis3d
[CVPR'19] JSIS3D: Joint Semantic-Instance Segmentation of 3D Point Clouds
cvpr deep-learning instance-segmentation point-cloud pytorch
Last synced: 31 Jul 2024
https://github.com/xiaoyu258/GeoProj
Blind Geometric Distortion Correction on Images Through Deep Learning
deep-learning distortion-correction pytorch
Last synced: 30 Jul 2024
https://github.com/eriklindernoren/Action-Recognition
Exploration of different solutions to action recognition in video, using neural networks implemented in PyTorch.
action-recognition pytorch video-classification
Last synced: 01 Aug 2024
https://github.com/jzlianglu/pykaldi2
Yet another speech toolkit based on Kaldi and PyTorch
horovod kaldi pykaldi pytorch speech-toolkit
Last synced: 17 Aug 2024
https://github.com/recursionpharma/gflownet
GFlowNet library specialized for graph & molecular data
deep-learning gflownet graph-neural-network pytorch
Last synced: 03 Aug 2024
https://github.com/mehmetgoren/feniks
A computer based NVR (Network Video Recorder) with AI capabilities
ai camera cctv face-recognition flv h265-hevc hls license-plate-recognition motion-detection nvr object-detection pytorch recoding rtmp rtsp streaming tensorflow video vms websocket
Last synced: 01 Aug 2024
https://github.com/phamquiluan/PubLayNet
ICDAR 2019: MaskRCNN on PubLayNet datasets. Paragraph detection, table detection, figure detection,...
document-layout-analysis figure-detection mask-rcnn object-detection paragraph-detection pretrained-models publaynet pytorch table-detection
Last synced: 08 Aug 2024
https://github.com/HEmile/storchastic
Stochastic Automatic Differentiation library for PyTorch.
gradient-estimators pytorch reparameterization stochastic-computation-graphs stochastic-deep-learning
Last synced: 05 Aug 2024
https://github.com/markovmodel/deeptime
Deep learning meets molecular dynamics.
autoencoder computational-biology computational-chemistry data-analysis deep-learning dimension-reduction machine-learning markov-model python pytorch tensorflow time-series
Last synced: 31 Jul 2024
https://github.com/Pilhyeon/BaSNet-pytorch
Official Pytorch Implementation of 'Background Suppression Network for Weakly-supervised Temporal Action Localization' (AAAI-20 Spotlight)
background-modeling deep-learning pytorch temporal-action-localization weakly-supervised-learning
Last synced: 02 Aug 2024
https://github.com/kangyeolk/Paint-by-Sketch
Stable Diffusion-based image manipulation method with a sketch and reference image
computer-vision diffusion-models image-editing image-generation image-manipulation pytorch stable-diffusion
Last synced: 31 Jul 2024
https://github.com/Js-Mim/mss_pytorch
Singing Voice Separation via Recurrent Inference and Skip-Filtering Connections - PyTorch Implementation. Demo:
deep-learning mss-pytorch music-source-separation pytorch recurrent-neural-networks skip-filtering-connections
Last synced: 03 Aug 2024
https://github.com/ahirner/pytorch-retraining
Transfer Learning Shootout for PyTorch's model zoo (torchvision)
benchmark pytorch transfer-learning
Last synced: 03 Aug 2024
https://github.com/balavenkatesh3322/audio-pretrained-model
A collection of Audio and Speech pre-trained models.
audio audio-processing caffe keras keras-models keras-tensorflow machine-learning mxnet neural-network pre-trained pre-trained-model pre-training python3 pytorch pytorch-models speech-recognition speech-to-text tensorflow tensorflow-models
Last synced: 01 Aug 2024
https://github.com/matajoh/fourier_feature_nets
Supplemental learning materials for "Fourier Feature Networks and Neural Volume Rendering"
computer-graphics computer-vision deep-learning pytorch
Last synced: 30 Aug 2024
https://github.com/pxiangwu/MotionNet
CVPR 2020, "MotionNet: Joint Perception and Motion Prediction for Autonomous Driving Based on Bird's Eye View Maps"
autonomous-driving cvpr2020 motion-prediction perception pytorch
Last synced: 31 Jul 2024
https://github.com/zudi-lin/pytorch_connectomics
PyTorch Connectomics: segmentation toolbox for EM connectomics
biomedical-image-processing computer-vision connectomics deep-learning microscopy neuroscience pytorch segmentation
Last synced: 02 Aug 2024
https://github.com/balavenkatesh3322/NLP-pretrained-model
A collection of Natural language processing pre-trained models.
deep-learning deep-neural-networks keras machine-learning model mxnet natural-language-processing neural-networks nlp nlp-machine-learning python python3 pytorch tensorflow text text-classification text-to-number text-to-speech
Last synced: 07 Aug 2024
https://github.com/shrubb/latent-pose-reenactment
The authors' implementation of the "Neural Head Reenactment with Latent Pose Descriptors" (CVPR 2020) paper.
avatar deep-learning face-reenactment facial-landmarks generative-model head-avatar head-reenactment landmark-detection pose-estimation pytorch self-supervised-learning talking-head voxceleb voxceleb2
Last synced: 03 Aug 2024
https://github.com/IlyaGusev/summarus
Models for automatic abstractive summarization
deep-learning machine-learning nlp pytorch summarization
Last synced: 01 Aug 2024
https://github.com/mithril-security/bastionlab
A simple framework for privacy-friendly data science collaboration
deep-learning eda multi-party privacy pytorch secure-enclave
Last synced: 02 Aug 2024
https://github.com/MrGemy95/visual-interaction-networks-pytorch
This's an implementation of deepmind Visual Interaction Networks paper using pytorch
computer-vision convolutional-neural-networks deep-learning deepmind machine-learning nerual-network physics pytorch video-prediction-models visual-interaction-networks
Last synced: 07 Aug 2024
https://github.com/atomicarchitects/equiformer_v2
[ICLR'24] EquiformerV2: Improved Equivariant Transformer for Scaling to Higher-Degree Representations
ai-for-science catalyst-design computational-chemistry computational-physics deep-learning drug-discovery e3nn equivariant-graph-neural-network equivariant-neural-networks force-fields geometric-deep-learning graph-neural-networks interatomic-potentials machine-learning materials-science molecular-dynamics pytorch
Last synced: 31 Jul 2024
https://github.com/wkcn/MobulaOP
A Simple & Flexible Cross Framework Operators Toolkit
cross-framework cupy deep-learning mxnet numpy operators pytorch scientific-computing
Last synced: 01 Aug 2024
https://github.com/fastaudio/fastaudio
🔊 Audio and fastai v2
audio deep-learning fastai gpu python pytorch
Last synced: 03 Aug 2024
https://github.com/j3soon/tbparse
Load tensorboard event logs as pandas DataFrames for scientific plotting; Supports both PyTorch and TensorFlow
machine-learning parser plot python pytorch reader tensorboard tensorboardx tensorflow
Last synced: 03 Aug 2024
https://github.com/ducha-aiki/pytorch-sift
PyTorch implementation of SIFT descriptor
cnn descriptor image-matching local-features pytorch sift
Last synced: 01 Aug 2024
https://github.com/AutoVision-cloud/SA-Det3D
[ICCVW-2021] SA-Det3D: Self-attention based Context-Aware 3D Object Detection
3d-deep-learning 3d-detection autonomous-driving object-detection point-cloud pytorch
Last synced: 31 Jul 2024
https://github.com/ShomyLiu/Neu-Review-Rec
A Toolkit for Neural Review-based Recommendation models with Pytorch.
daml deepconn narre pytorch recommendation-system review-based
Last synced: 03 Aug 2024
https://github.com/sangyx/d2l-torch
《动手学深度学习》 PyTorch 版本
chinese deep-learning jupyter-notebook python pytorch
Last synced: 31 Jul 2024
https://github.com/benedekrozemberczki/role2vec
A scalable Gensim implementation of "Learning Role-based Graph Embeddings" (IJCAI 2018).
deep-learning deepwalk gensim graph-embedding graph-neural-network graph-wavelet implicit-factorization machine-learning network-embedding network-science node-embedding node2vec pytorch representation-learning research sklearn struc2vec tensorflow weisfeiler-lehman word2vec
Last synced: 01 Aug 2024
https://github.com/VSainteuf/pytorch-psetae
PyTorch implementation of the model presented in "Satellite Image Time Series Classification with Pixel-Set Encoders and Temporal Self-Attention"
agriculture computer-vision cvpr cvpr2020 deep-learning earth-observation machine-learning pytorch remote-sensing satellite-image self-attention spatio-temporal time-series-classification transformer-architecture
Last synced: 03 Aug 2024
https://github.com/HennyJie/BrainGB
Officially Accepted to IEEE Transactions on Medical Imaging (TMI, IF: 11.037) - Special Issue on Geometric Deep Learning in Medical Imaging.
ai4health brain-networks gcn graph-convolutional-networks graph-neural-networks network-embedding pytorch
Last synced: 02 Aug 2024
https://github.com/geekjr/quickai
QuickAI is a Python library that makes it extremely easy to experiment with state-of-the-art Machine Learning models.
ai artificial-intelligence bert deep-learning dl easy-to-use fast gpt gpt-neo huggingface-transformers ml neural-network nlp object-detection python pytorch quickai research tensorflow2 yolo
Last synced: 02 Aug 2024
https://github.com/openclimatefix/graph_weather
PyTorch implementation of Ryan Keisler's 2022 "Forecasting Global Weather with Graph Neural Networks" paper (https://arxiv.org/abs/2202.07575)
forecasting-models graph-neural-networks pytorch weather
Last synced: 31 Jul 2024
https://github.com/CleanPegasus/TensorDash
TensorDash is an application that lets you remotely monitor your deep learning model's metrics and notifies you when your model training is completed or crashed.
android deep-learning fastai java keras machine-learning pypi python python-package pytorch remote-monitoring tensorflow
Last synced: 03 Aug 2024
https://github.com/Chris-hughes10/pytorch-accelerated
A lightweight library designed to accelerate the process of training PyTorch models by providing a minimal, but extensible training loop which is flexible enough to handle the majority of use cases, and capable of utilizing different hardware options with no code changes required. Docs: https://pytorch-accelerated.readthedocs.io/en/latest/
deep-learning huggingface-accelerate huggingface-examples pytorch pytorch-tutorial timm-examples training
Last synced: 05 Aug 2024
https://github.com/SforAiDl/vformer
A modular PyTorch library for vision transformer models
Last synced: 03 Aug 2024
https://github.com/JieShibo/PETL-ViT
[ICCV 2023] Binary Adapters, [AAAI 2023] FacT, [Tech report] Convpass
parameter-efficient-tuning pytorch transfer-learning visual-prompting
Last synced: 01 Aug 2024
https://github.com/jayleicn/TVQA
[EMNLP 2018] PyTorch code for TVQA: Localized, Compositional Video Question Answering
Last synced: 01 Aug 2024
https://github.com/pyg-team/pyg-lib
Low-Level Graph Neural Network Operators for PyG
deep-learning geometric-deep-learning graph-convolutional-networks graph-neural-networks pytorch
Last synced: 31 Jul 2024
https://github.com/wimlds-trojmiasto/detect-waste
AI to Combat Environmental Pollution - detecting plastic waste in the environment to combat environmental pollution and promote circular economy (Deep Learning, PyTorch)
cnn deep-learning detr efficientdet fastrcnn litter maskrcnn neural-networks object-detection python pytorch trash waste-detection
Last synced: 03 Aug 2024
https://github.com/huggingface/diffusion-fast
Faster generation with text-to-image diffusion models.
diffusers diffusion-models pytorch sdxl text-to-image-generation
Last synced: 01 Aug 2024
https://github.com/uvipen/SSD-pytorch
SSD: Single Shot MultiBox Detector pytorch implementation focusing on simplicity
computer-vision deep-learning deep-neural-networks deeplearning detection detector neural-networks object-detection pytorch ssd ssdlite
Last synced: 07 Aug 2024
https://github.com/antao97/UnsupervisedPointCloudReconstruction
Experiments on unsupervised point cloud reconstruction.
auto-encoder feature-learning point-cloud pytorch unsupervised-learning
Last synced: 31 Jul 2024
https://github.com/shinianzhihou/ChangeDetection
A framework for change detection using PyTorch
change-detection pytorch remote-sensing siamese unet
Last synced: 03 Aug 2024
https://github.com/araffin/srl-zoo
State Representation Learning (SRL) zoo with PyTorch - Part of S-RL Toolbox
autoencoder deep-learning forward-model inverse-model neural-network pytorch reinforcement-learning representation-learning srl state-representation-learning vae
Last synced: 31 Jul 2024
https://github.com/GuoleiSun/CountSeg
Official code for "Object counting and instance segmentation with image-level supervision", in CVPR 2019 and TPAMI 2020
cvpr2019 instance-segmentation object-counting pytorch tpami2020 weakly-supervised-learning
Last synced: 01 Aug 2024
https://github.com/IDSIA/modern-srwm
Official repository for the paper "A Modern Self-Referential Weight Matrix That Learns to Modify Itself" (ICML 2022 & NeurIPS 2021 Deep RL Workshop) and "Accelerating Neural Self-Improvement via Bootstrapping" (ICLR 2023 Workshop)
fast-weight-programmers fast-weights pytorch reinforcement-learning rnn-based-few-shot-learning self-referential-learning self-referential-weight-matrix torchbeast transformers
Last synced: 02 Aug 2024
https://github.com/mostafaelaraby/wavegan-pytorch
Pytorch Implementation of wavegan model to generate audio
ai-music gan music-generation pytorch wavegan wavegan-pytorch
Last synced: 04 Aug 2024
https://github.com/Project-AgML/AgML
AgML is a centralized framework for agricultural machine learning. AgML provides access to public agricultural datasets for common agricultural deep learning tasks, with standard benchmarks and pretrained models, as well the ability to generate synthetic data and annotations.
agriculture computer-vision dataset deep-learning image-classification object-detection pytorch semantic-segmentation synthetic-data
Last synced: 03 Aug 2024
https://github.com/lukas-blecher/AutoMask
Automatic Rotoscoping/Masking tool powered with Machine Learning for Blender (WIP)
blender blender-addon blender-python deep-learning machine-learning mask-segment masking motion-tracking pytorch rotoscoping
Last synced: 03 Aug 2024
https://github.com/benjs/nfnets_pytorch
Pre-trained NFNets with 99% of the accuracy of the official paper "High-Performance Large-Scale Image Recognition Without Normalization".
deepmind nfnets pretrained-models pytorch sgd-adc weight-standardization
Last synced: 05 Aug 2024
https://github.com/greydanus/mnist1d
A 1D analogue of the MNIST dataset for measuring spatial biases and answering Science of Deep Learning questions.
convnet dataset machine-learning neural-networks pytorch research
Last synced: 31 Jul 2024
https://github.com/DavexPro/pytorch-pose-estimation
PyTorch Implementation of Realtime Multi-Person Pose Estimation project.
computer-vision cvpr-2017 deep-learning human-pose-estimation python3 pytorch
Last synced: 07 Aug 2024
https://github.com/Yachay-AI/byt5-geotagging
Confidence and Byt5 - based geotagging model predicting coordinates from text alone.
coordinates deep-learning geo-location geotagging machine-learning neural-network nlp nlp-machine-learning python pytorch transformers
Last synced: 01 Aug 2024
https://github.com/bfshi/AbSViT
Official code for "Top-Down Visual Attention from Analysis by Synthesis" (CVPR 2023 highlight)
attention classification cvpr pytorch segmentation vision-transformer
Last synced: 02 Aug 2024
https://github.com/floydhub/floyd-cli
Command line tool for FloydHub - the fastest way to build, train, and deploy deep learning models
ai artificial-intelligence deep-learning floyd floyd-cli floydhub gpu machine-learning python pytorch tensorflow
Last synced: 31 Jul 2024
https://github.com/floydhub/dockerfiles
Deep Learning Dockerfiles
deep-learning docker pytorch tensorflow torch
Last synced: 01 Aug 2024
https://github.com/Ai-trainee/Traffic-Sign-Recognition-PyQt5-YOLOv5-GUI
This YOLOv5🚀😊 GUI road sign system uses MySQL💽, PyQt5🎨, PyTorch, CSS🌈. It has modules for login🔑, YOLOv5 setup📋, sign recognition🔍, database💾, and image processing🖼️. It supports diverse inputs, model switching, and enhancements like mosaic and mixup📈.
css mysql-database pyqt5 pytorch tensorrt-engine yolov5 yolov5-gui
Last synced: 02 Aug 2024
https://github.com/hou-yz/MVDet
[ECCV 2020] Codes and MultiviewX dataset for "Multiview Detection with Feature Perspective Transformation".
dataset detection detector multiview pedestrian-detection pytorch
Last synced: 31 Jul 2024
https://github.com/isLinXu/YOLOv8_Efficient
🚀Simple and efficient use for Ultralytics yolov8🚀
deeplearning pytorch ultralytics yolov8
Last synced: 02 Aug 2024
https://github.com/p-ranav/PhotoLab
AI-Powered Photo Editor (Python, PyQt6, PyTorch)
animegan colorization cuda human-segmentation interactive nuitka numpy opencv photo-editor pillow portrait-mode pyqt6 pyqt6-desktop-application python python3 pytorch scikit-image spot-removal stacking
Last synced: 01 Aug 2024
https://github.com/paul007pl/VRCNet
[CVPR 2021 Oral] Variational Relational Point Completion Network
mvp-dataset partial-point-clouds point-cloud-completion pytorch
Last synced: 31 Jul 2024
https://github.com/undark-lab/swyft
A system for scientific simulation-based inference at scale.
likelihood-free-inference machine-learning marginal-neural-ratio-estimation neural-ratio-estimation parameter-estimation python pytorch simulation-based-inference truncated-neural-ratio-estimation
Last synced: 31 Jul 2024
https://github.com/ozanciga/diffusion-for-beginners
denoising diffusion models, as simple as possible
dall-e diffusion imagen midjourney pytorch scheduler stable-diffusion
Last synced: 03 Aug 2024
https://github.com/aramis-lab/clinicadl
Framework for the reproducible processing of neuroimaging data with deep learning methods
alzheimer-disease brain-imaging convolutional-neural-network deep-learning medical-imaging neuroimaging python pytorch
Last synced: 02 Aug 2024
https://github.com/CDOTAD/AlphaGAN-Matting
This project is an unofficial implementation of AlphaGAN: Generative adversarial networks for natural image matting published at the BMVC 2018
Last synced: 03 Aug 2024
https://github.com/codeslake/Color_Transfer_Histogram_Analogy
[CGI 2020] Official PyTorch Implementation for "Deep Color Transfer using Histogram Analogy"
color-transfer deep-learning histogram-analogy pytorch
Last synced: 03 Aug 2024
https://github.com/chrisby/torchMTL
A lightweight module for Multi-Task Learning in pytorch.
deep-learning framework machine-learning mtl multi-task-learning python3 pytorch
Last synced: 03 Aug 2024
https://github.com/jingxil/Neural-Decision-Forests
An implementation of the Deep Neural Decision Forests in PyTorch
Last synced: 31 Jul 2024
https://github.com/mrdbourke/mac-ml-speed-test
A few quick scripts focused on testing TensorFlow/PyTorch/Llama 2 on macOS.
apple-silicon benchmark deep-learning llama2 llamacpp llm m1 m1-mac m2-mac m3-mac machine-learning macos metal metal-performance-shaders ml mps pytorch speedtest tensorflow2
Last synced: 06 Aug 2024
https://github.com/sanyam5/arc-pytorch
The first public PyTorch implementation of Attentive Recurrent Comparators
computer-vision deep-learning deep-neural-networks machine-learning pytorch recurrent-neural-networks
Last synced: 07 Aug 2024
https://github.com/fcakyon/small-object-detection-benchmark
icip2022 paper: sahi benchmark on visdrone and xview datasets using fcos, vfnet and tood detectors
benchmark coco fcos icip2022 mmdetection object-detection pytorch sahi small-object-detection tood vfnet visdrone xview
Last synced: 03 Aug 2024
https://github.com/jla524/fromthetensor
From the Tensor to Stable Diffusion, a rough outline for a 9 week course.
deep-learning pytorch tinygrad
Last synced: 04 Aug 2024
https://github.com/szq0214/Un-Mix
Un-Mix: Rethinking Image Mixtures for Unsupervised Visual Representation Learning.
image-mixture pytorch self-supervised-learning smooth-decision-boundaries unsupervised-learning
Last synced: 02 Aug 2024
https://github.com/ViCCo-Group/THINGSvision
Python package for extracting representations from state-of-the-art computer vision models
alignment cognitive-science computer-vision deep-learning neural-networks pytorch representations tensorflow
Last synced: 03 Aug 2024
https://github.com/Graph-COM/GSAT
[ICML 2022] Graph Stochastic Attention (GSAT) for interpretable and generalizable graph learning.
deep-learning graph-neural-networks interpretability interpretable-machine-learning pytorch xai
Last synced: 08 Aug 2024
https://github.com/dmmiller612/lecture-summarizer
Lecture summarization with BERT
bert bert-model extractive-summarization flask pytorch summarization
Last synced: 01 Aug 2024
https://github.com/ucbrise/flor
🌻 FlorFlow: Flor, now with Dataflow
airflow build dag deep-learning flor hindsight logger logging machine-learning ml pytorch tensorboard vldb
Last synced: 01 Aug 2024
https://github.com/kennethnym/infinifi
infinifi plays gentle lofi music in the background indefinitely
artificial-intelligence audiocraft lo-fi lo-fi-radio machine-learning musicgen pytorch
Last synced: 31 Jul 2024
https://github.com/foivospar/NED
PyTorch implementation for NED (CVPR 2022). It can be used to manipulate the facial emotions of actors in videos based on emotion labels or reference styles.
3dmm cvpr2022 deepfakes emotion expression-manipulation face face-edit gan image-generation pytorch talking-head
Last synced: 03 Aug 2024
https://github.com/jason-li-831202/Vehicle-CV-ADAS
The project can achieve FCWS, LDWS, and LKAS functions solely using only visual sensors. using YOLOv5 / YOLOv5-lite / YOLOv6 / YOLOv7 / YOLOv8 / YOLOv9 / EfficientDet and Ultra-Fast-Lane-Detection-v2 .
bytetrack efficientdet fcws lane-detection ldws lkas object-detection onnxruntime python pytorch tensorrt ultra-fast-lane-detection-v2 yolov10 yolov5 yolov5-lite yolov6 yolov7 yolov8 yolov9
Last synced: 02 Aug 2024
https://github.com/DeepRank/deeprank
This repository has been integrated in https://github.com/DeepRank/deeprank2
3d-cnn docking protein-protein-interaction pytorch
Last synced: 08 Aug 2024
https://github.com/haruishi43/equilib
🌎→🗾Equirectangular (360/panoramic) image processing library for Python with minimal dependencies only using Numpy and PyTorch
360-degree camera cubemap-to-equirectangular equirectangular-images equirectangular-panorama equirectangular-projection numpy panorama perspective-projection python pytorch vr
Last synced: 30 Jul 2024
https://github.com/kashif/firedup
Clone of OpenAI's Spinning Up in PyTorch
deep-learning pytorch reinforcement-learning spinningup
Last synced: 02 Aug 2024
https://github.com/forestagostinelli/DeepCubeA
Code for DeepCubeA, a Deep Reinforcement Learning algorithm that can learn to solve the Rubik's cube.
astar-search deep-learning deep-reinforcement-learning deepcube deepcubea python3 pytorch reinforcement-learning rubiks-cube search
Last synced: 09 Aug 2024
https://github.com/erfanzar/EasyDeL
EasyDeL is an OpenSource Library to make your training faster and more Optimized With cool Options for training and serving Both in Python And Mojo🔥
easydel flax gpt jax machine-learning mojo nlp optax pytorch transformers
Last synced: 03 Aug 2024
https://github.com/eambutu/snail-pytorch
Implementation of "A Simple Neural Attentive Meta-Learner" (SNAIL, https://arxiv.org/pdf/1707.03141.pdf) in PyTorch
Last synced: 30 Jul 2024
https://github.com/guillaume-chevalier/Linear-Attention-Recurrent-Neural-Network
A recurrent attention module consisting of an LSTM cell which can query its own past cell states by the means of windowed multi-head attention. The formulas are derived from the BN-LSTM and the Transformer Network. The LARNN cell with attention can be easily used inside a loop on the cell state, just like any other RNN. (LARNN)
attention-is-all-you-need attention-mechanism attention-model lstm pytorch recurrent-neural-networks rnn
Last synced: 01 Aug 2024
https://github.com/hav4ik/Hydra
Multi-Task Learning Framework on PyTorch. State-of-the-art methods are implemented to effectively train models on multiple tasks.
deep-learning multi-objective-optimization multi-task-learning neural-architecture-search pytorch
Last synced: 03 Aug 2024
https://github.com/aidos-lab/pytorch-topological
A topological machine learning framework based on PyTorch
deep-learning pytorch topological-data-analysis topological-machine-learning
Last synced: 02 Aug 2024
https://github.com/LeapLabTHU/Pseudo-Q
[CVPR 2022] Pseudo-Q: Generating Pseudo Language Queries for Visual Grounding
computer-vision cvpr2022 deep-learning multimodal-deep-learning pytorch vision-and-language visual-grounding
Last synced: 01 Aug 2024
https://github.com/nihalsid/ViewAL
[CVPR'20] Implementation for the paper "ViewAL: Active Learning with Viewpoint Entropy for Semantic Segmentation"
active-learning deep-learning pytorch semantic-segmentation
Last synced: 31 Jul 2024
https://github.com/Na-Z/attMPTI
[CVPR 2021] Few-shot 3D Point Cloud Semantic Segmentation
3d 3d-segmentation few-shot-learning point-cloud pytorch
Last synced: 31 Jul 2024
https://github.com/TIGER-AI-Lab/ImagenHub
A one-stop library to standardize the inference and evaluation of all the conditional image generation models. (ICLR 2024)
deep-learning diffusion-models generative-ai generative-art image-editing image-generation pytorch stable-diffusion
Last synced: 06 Aug 2024
https://github.com/AllenXiangX/SnowflakeNet
(TPAMI 2023) Snowflake Point Deconvolution for Point Cloud Completion and Generation with Skip-Transformer
3dvision deep-learning iccv2021 point-cloud-completion pytorch tpami2023 transformers
Last synced: 03 Aug 2024