A Pytorch implementation of Sparsely-Gated Mixture of Experts, for massively increasing the parameter count of language models

LibKGE - A knowledge graph embedding library for reproducible research

Files and resources for the cyberdeck RPI DEV (Raspberry Pi Development Platform). This includes .stl files, kicad files, BOM, instructions, and sample code.

A Large-Scale Few-Shot Relation Extraction Dataset

Convolutional 2D Knowledge Graph Embeddings resources

R1-searcher: Incentivizing the Search Capability in LLMs via Reinforcement Learning

A tensorflow implementation of RippleNet

This is the official github repo of Think-on-Graph (ICLR 2024). If you are interested in our work or willing to join our research team in Shenzhen, please feel free to contact us by email (xuchengj…

A Pytorch Implementation of "Attention is All You Need" and "Weighted Transformer Network for Machine Translation"

A tensorflow implementation of Knowledge Graph Convolutional Networks

Official Implementation of NeurIPS 2024 paper "G-Retriever: Retrieval-Augmented Generation for Textual Graph Understanding and Question Answering""

Official Implementation of ICLR 2024 paper: "Reasoning on Graphs: Faithful and Interpretable Large Language Model Reasoning"

Graph-R1: Towards Agentic GraphRAG Framework via End-to-end Reinforcement Learning

Implementation of ST-Moe, the latest incarnation of MoE after years of research at Brain, in Pytorch

MindMap: Knowledge Graph Prompting Sparks Graph of Thoughts in Large Language Models

Implementation of Soft MoE, proposed by Brain's Vision team, in Pytorch

Reinforcement Knowledge Graph Reasoning for Explainable Recommendation

Code and Checkpoints for "Generate rather than Retrieve: Large Language Models are Strong Context Generators" in ICLR 2023.

MoH: Multi-Head Attention as Mixture-of-Head Attention

[ICLR 2025] MoE++: Accelerating Mixture-of-Experts Methods with Zero-Computation Experts

Triton-based implementation of Sparse Mixture of Experts.

Federated Learning for LLMs.

Convert image/gif/video to ascii art.

CoDEx: A set of knowledge graph Completion Datasets Extracted from Wikidata and Wikipedia

Exploring large language models for knowledge graph completion. ICASSP 2025

Official repository of "Can Language Models Solve Graph Problems in Natural Language?". NeurIPS 2023 (Spotlight)

[NeurIPS'24] Knowledge Graph Fine-Tuning using LLMs

We want to try and evaluate LLMs using Knowledge Graphs