jaketae

storyteller

Multimodal AI Story Teller, built with Stable Diffusion, GPT, and neural text-to-speech

koclip

KoCLIP: Korean port of OpenAI CLIP, in Flax

deep-malware-detection

A neural approach to malware detection in portable executables

wordwise

N-gram keyword extraction using spaCy and pretrained language models

ensemble-transformers

Ensembling Hugging Face transformers made easy

g-mlp

PyTorch implementation of Pay Attention to MLPs

fnet

PyTorch implementation of FNet: Mixing Tokens with Fourier transforms

param-share-transformer

PyTorch implementation of Lessons on Parameter Sharing across Layers in Transformers

alibi

PyTorch implementation of Train Short, Test Long: Attention with Linear Biases Enables Input Length Extrapolation

mlp-mixer

PyTorch implementation of MLP-Mixer: An all-MLP Architecture for Vision

jaketae.github.io

Personal blog with Jupyter notebooks

realformer

PyTorch implementation of RealFormer: Transformer Likes Residual Attention

lm-identifier

A toolkit for identifying pretrained language models from potentially AI-generated text

tupe

PyTorch implementation of Rethinking Positional Encoding in Language Pre-training

vit-breast-cancer

Transfer learning pretrained vision transformers for breast histopathology

graph-neural-ode

Graph neural ordinary differential equations

transformers

🤗 Transformers: State-of-the-art Natural Language Processing for Pytorch, TensorFlow, and JAX.

accelerate

🚀 A simple way to train and use PyTorch models with multi-GPU, TPU, mixed-precision

datasets

🤗 Fast, efficient, open-access datasets and evaluation metrics in PyTorch, TensorFlow, NumPy and Pandas

diffusers

🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch

fairseq

Facebook AI Research Sequence-to-Sequence Toolkit written in Python.

H3

Language Modeling with the H3 State Space Model

hillhouse

Reimagining the newly constructed Hillhouse Avenue of Yale University

ivy

The Unified Machine Learning Framework

jaketae

nn-svd

Neural network compression with SVD

promptsource

Toolkit for collecting and applying prompts

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

speechbrain

A PyTorch-based Speech Toolkit

TTS

🐸💬 - a deep learning toolkit for Text-to-Speech, battle-tested in research and production