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)

sockeye

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

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.

caver

Caver: a toolkit for multilabel text classification.

steamboat-bio

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

audio-flamingo

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

open-namm

An open source implementation of the paper: "AN EVOLVED UNIVERSAL TRANSFORMER MEMORY"

swin-transformer-pytorch

Implementation of the Swin Transformer in PyTorch.

selfextend

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