xiaotingsong

The-Art-of-Linear-Algebra

Graphic notes on Gilbert Strang's "Linear Algebra for Everyone"

pykan

Kolmogorov Arnold Networks

Swin-Transformer

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".

mamba

Mamba SSM architecture

Megatron-LM

Ongoing research training transformer models at scale

ChatRWKV

ChatRWKV is like ChatGPT but powered by RWKV (100% RNN) language model, and open source.

DiT

Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"

earthworm

Learning English through the method of constructing sentences with conjunctions

Vim

[ICML 2024] Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model

releasing-research-code

Tips for releasing research code in Machine Learning (with official NeurIPS 2020 recommendations)

Semi-supervised-learning

A Unified Semi-Supervised Learning Codebase (NeurIPS'22)

Time-LLM

[ICLR 2024] Official implementation of " 🦙 Time-LLM: Time Series Forecasting by Reprogramming Large Language Models"

mamba.py

A simple and efficient Mamba implementation in pure PyTorch and MLX.

soft-dtw

Python implementation of soft-DTW.

earth-forecasting-transformer

Official implementation of Earthformer

Med-BERT

Med-BERT, contextualized embedding model for structured EHR data

TravelPlanner

[ICML'24 Spotlight] "TravelPlanner: A Benchmark for Real-World Planning with Language Agents"

simba

Simba

S-D-Mamba

Code for "Is Mamba Effective for Time Series Forecasting?"

pytorch_ehr

source codes based on PyTorch to analyze EHR

LLMScenarioEval

Scenario-based Evaluation dataset for LLM (beta)

Imbalance-VLM

BEHRT

Code for BEHRT: Transformer for Electronic Health Records

New-Grad-Internships

👋 Howdy New Grads and Interns 🎉! We've put together a collection of jobs for SWE, Quant and Data Science Roles! 🚀

ohmnet

OhmNet: Representation learning in multi-layer graphs

mamba360

State Space Models

cehr-bert

CEHR-BERT: Incorporating temporal information from structured EHR data to improve prediction tasks

Warpformer

GNPP-code

The implementation of GNPP model.

Logic_Point_Processes_ICLR