Attention Model Python Packages

whisper-timestamped

Multilingual Automatic Speech Recognition with word-level timestamps and confidence

zetascale

Build high-performance AI models with modular building blocks

rl4co

A PyTorch library for all things Reinforcement Learning (RL) for Combinatorial Optimization (CO)

swin-transformer-pytorch

Implementation of the Swin Transformer in PyTorch.

sockeye

Sequence-to-sequence framework with a focus on Neural Machine Translation based on PyTorch

steamboat-bio

Steamboat is an interpretable framework to study multiscale cellular interactions in tissues

seq2seqsharp

Seq2SeqSharp is a tensor based fast & flexible deep neural network framework written by .NET (C#). It has many highlighted features, such as automatic differentiation, different network types (Transformer, LSTM, BiLSTM and so on), multi-GPUs supported, cross-platforms (Windows, Linux, MacOS), multimodal model for text and images and so on.

audio-flamingo

Implementation of the model "AudioFlamingo" from the paper: "Audio Flamingo: A Novel Audio Language Model with Few-Shot Learning and Dialogue Abilities"

caver

Multi-label Text Classification Toolkit

open-namm

Paper - Pytorch

selfextend

Implementation of SelfExtend from the paper "LLM Maybe LongLM: Self-Extend LLM Context Window Without Tuning" from Pytorch and Zeta