Earthring

JaxTransformer

This repository demonstrates how to build a Decoder-Only Transformer with Multi-Query Attention in JAX.

Awesome-World-Model

Collect some World Models for Autonomous Driving papers.

RecSysPapers

推荐/广告/搜索领域工业界经典以及最前沿论文集合。A collection of industry classics and cutting-edge papers in the field of recommendation/advertising/search.

bridge_env

Contract Bridge Environment package in Python

diamond

DIAMOND (DIffusion As a Model Of eNvironment Dreams) is a reinforcement learning agent trained in a diffusion world model.

DocBench

DocBench: A Benchmark for Evaluating LLM-based Document Reading Systems

ml-visuals

🎨 ML Visuals contains figures and templates which you can reuse and customize to improve your scientific writing.

Craftax

(Crafter + NetHack) in JAX. ICML 2024 Spotlight.

ShiArthur03

causal-learn

Causal Discovery in Python. It also includes (conditional) independence tests and score functions.

granular

Fast dataset format and loader

awesome-model-based-RL

A curated list of awesome model based RL resources (continually updated)

clash-core

backup of clash core

TimeSiam

Deductive-Beam-Search

[COLM'24] "Deductive Beam Search: Decoding Deducible Rationale for Chain-of-Thought Reasoning"

CarDreamer

World Model based Autonomous Driving Platform in CARLA :car:

Qwen2.5

Qwen2.5 is the large language model series developed by Qwen team, Alibaba Cloud.

SHINE

Official code for: "SHINE: Shielding Backdoors in Deep Reinforcement Learning"

RigorLLM

Implementation for "RigorLLM: Resilient Guardrails for Large Language Models against Undesired Content"

Omost

Your image is almost there!

HarmonyDream

Code release for "HarmonyDream: Task Harmonization Inside World Models" (ICML 2024), https://arxiv.org/abs/2310.00344

HelmFluid

About code release of "HelmFluid: Learning Helmholtz Dynamics for Interpretable Fluid Prediction", ICML 2024. https://arxiv.org/pdf/2310.10565

Transolver

About code release of "Transolver: A Fast Transformer Solver for PDEs on General Geometries", ICML 2024 Spotlight. https://arxiv.org/abs/2402.02366

Uni-RLHF-Platform

Uni-RLHF platform for "Uni-RLHF: Universal Platform and Benchmark Suite for Reinforcement Learning with Diverse Human Feedback" (ICLR2024)

Clean-Offline-RLHF

Offline RLHF codebase implementation for "Uni-RLHF: Universal Platform and Benchmark Suite for Reinforcement Learning with Diverse Human Feedback" (ICLR2024)

Multi-Embedding

About Code Release for "On the Embedding Collapse When Scaling Up Recommendation Models" (ICML 2024)

sbx

SBX: Stable Baselines Jax (SB3 + Jax)

iVideoGPT

Official repository for "iVideoGPT: Interactive VideoGPTs are Scalable World Models" (NeurIPS 2024), https://arxiv.org/abs/2405.15223

hok_env

Honor of Kings AI Open Environment of Tencent

General-World-Models-Survey