Vector Institute Implementation Catalog

A curated collection of high-quality AI implementations developed by researchers and engineers at the Vector Institute

73

Implementations

7

Years of Research

Browse Implementations by Type¶

applied-researchbootcamptool

Library for handling atomistic graph datasets focusing on transformer-based implementations, with utilities for training various models, experimenting with different pre-training tasks, and a suite of pre-trained models with huggingface integrations

AtomFormer SchNet TokenGT

Datasets: S2EF Datasets Misc. Atomistic Graph Datasets

A comprehensive framework for Knowledge Graph Retrieval Augmented Generation (KG-RAG).

KG-RAG GraphRAG

Datasets: SEC 10-Q

A toolkit to download, augment, and benchmark Open-PMC data

PMC Data Extraction

Datasets: PubMed Central HuggingFace PMC Dataset

Cite Paper

A repository for social bias mitigation in LLMs using machine unlearning

Negation via Task Vectors PCGU

Datasets: BBQ Stereoset RedditBias

Cite Paper

A repository with implementation of anomaly detection techniques

Logistic Regression (Supervised) Random Forest (Supervised) XGBoost (Supervised) CatBoost (Supervised) Light GBM (Supervised) TabNet (Supervised and Semi-supervised) Autoencoder (AE) (Unsupervised) Isolation Forest (Unsupervised)

Datasets: Bank Account Fraud Detection DGraph dataset MVTec dataset UCSD Anomaly Detection Dataset UCF Crime Dataset

A repository with implementations of recommender systems

Matrix Factorization Collaborative Filtering Content-Based Filtering Sequence Aware Recommender Systems Session-Based Recommender Systems Knowledge Graph-Based Recommender Systems

A repository with implementations of privacy-enhancing techniques for machine learning

Differential Privacy (tensorflow_privacy) PATE Membership Inference Attacks Horizontal Federated Learning Vertical Federated Learning Homomorphic Encryption

A repository with reference implementations of self-supervised learning techniques

Internal Contrastive Learning (ICL) + Latent Outlier Exposure (LOE) SimMTM TabRet Data2Vec

Datasets: STL-10 Beijing PM 2.5

A repository with demos for various diffusion models for tabular and time series data

TabDDPM TabSyn ClavaDDPM CSDI TSDiff

Datasets: Physionet Challenge 2012 Electricity dataset (UCI Machine Learning Repository)

A repository with reference implementations for deploying AI models in production environments, focusing on best practices and cloud-native solutions.

AWS GCP

A repository reference implementations for retrieval-augmented generation

Web Search Document Search SQL Search Cloud Search PubMed QA RAG Evaluation

Datasets: PubMed Banking Dataset - Marketing Targets

A repository with implementations advanced fine-tuning techniques and approaches to enhance Large Language Model performance, reduce their computational cost, with a focus on alignment with human values

FSDP DDP Instruction Tuning PEFT Quantization Supervised Fine-tuning

Datasets: SAMSum dataset TweetEval

A framework for fine-tuning retrieval-augmented generation (RAG) systems.

Basic fine-tuning with FL RA-DIT

Efficient LLM inference on Slurm clusters using vLLM.

A toolkit for research on multimodal representation learning

Contrastive Pretraining I-JEPA

Datasets: ImageNet LibriSpeech RGB-D

A flexible, modular, and easy to use library to facilitate federated learning research and development in healthcare settings

FedAvg FedOpt FedProx SCAFFOLD MOON FedDG-GA FLASH FedPM Personal FL FedBN FedPer FedRep Ditto MR-MTL APFL PerFCL FENDA-FL FENDA+Ditto

Browse Implementations by Type¶

atomgen

kg-rag

pmc-data-extraction

bias-mitigation-unlearning

anomaly-detection

recommender-systems

privacy-enhancing-techniques

self-supervised-learning

diffusion-models

ai-deployment

retrieval-augmented-generation

finetuning-and-alignment

fed-rag

vector-inference

mmlearn

fl4health