Fast Matrix Multiplications for Lookup Table-Quantized LLMs

This repository is used for advertising PhD recruitment opportunities. Contributions are welcome!

Simple, minimal implementation of the Mamba SSM in one file of PyTorch.

Efficient implementations of state-of-the-art linear attention models in Pytorch and Triton

NAACL '24 (Best Demo Paper RunnerUp) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inference

List of companies offering Machine learning and Data Science internships

PyTorch(1.6+) implementation of https://github.com/kang205/SASRec

Open Bandit Pipeline: a python library for bandit algorithms and off-policy evaluation

🎓 无需编写任何代码即可轻松创建漂亮的学术网站 Easily create a beautiful academic résumé or educational website using Hugo and GitHub. No code.

Network Analysis in Python

TikZ library for drawing Bayesian networks, graphical models and (directed) factor graphs in LaTeX.

Pytorch implementation of convolutional neural network visualization techniques

Visualization toolkit for neural networks in PyTorch! Demo -->

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…

Message Passing Neural Networks for Molecule Property Prediction

Collection of Summer 2025 tech internships!

Python package for graph neural networks in chemistry and biology

Python package built to ease deep learning on graph, on top of existing DL frameworks.

An Awesome list of long-term visas

astir | Automated cell identity from single-cell multiplexed imaging and proteomics 🖥🔬✨

A repository for CSSU Members to share collaborate on the job search

Cookiecutter python package using Poetry, mypy, black, isort, autoflake, pytest, mkdocs, and GitHub Actions

Play with neural networks!

Deep probabilistic analysis of single-cell and spatial omics data