Search Packages

mambavision

[CVPR 2025] Official PyTorch Implementation of MambaVision: A Hybrid Mamba-Transformer Vision Backbone

keras-vision

Porting vision models to Keras 3 for easily accessibility. Contains MobileViT v1, MobileViT v2, fastvit

fastervit

FasterViT: Fast Vision Transformers with Hierarchical Attention

crabnet

Predict materials properties using only the composition information!

gcvit

Tensorflow 2.0 Implementation of GCViT: Global Context Vision Transformer

gsa-pytorch

Global Self-attention Network (GSA) - Pytorch

deberta

The implementation of DeBERTa

keyatten

KeyAtten: Attention-based Zero-Shot Keyword & Keyphrase Extraction

self-attention-cv

Implementation of various self-attention mechanisms focused on computer vision. Ongoing repository.

perceiver-io-pytorch

Unofficial implementation of Perceiver IO

trphysx

Transformers for modeling physical systems

transformerx

Flexible Python library providing building blocks (layers) for reproducible Transformers research (Tensorflow ✅, Pytorch 🔜, and Jax 🔜)

arenets

Tensorflow-based framework which lists implementation of conventional neural network models (CNN, RNN-based) for Relation Extraction classification tasks as well as API for custom model implementation

rnn4ie

Chinese Information Extraction Toolkit。中文信息抽取工具

fixedvit

fixed version of fastervit implementation https://github.com/NVlabs/FasterViT

correctedvit

[ICLR 2024] Official PyTorch implementation of FasterViT: Fast Vision Transformers with Hierarchical Attention

fastervitfixed

fixed version of fastervit implementation https://github.com/NVlabs/FasterViT

graphattentionnetworks

This package is a Tensorflow2/Keras implementation for Graph Attention Network embeddings and also provides a Trainable layer for Multihead Graph Attention.