Tianwei Ni

pomdp-baselines. Simple (but often Strong) Baselines for POMDPs in PyTorch, ICML 2022

OrganSegRSTN_PyTorch. PyTorch implementation of OrganSegRSTN - CVPR 2018

Memory-RL. When Do Transformers Shine in RL? Decoupling Memory from Credit Assignment, NeurIPS 2023 (oral)

Elastic-Boundary-Projection. Elastic Boundary Projection for 3D Medical Image Segmentation - CVPR 2019

f-IRL. Inverse Reinforcement Learning via State Marginal Matching, CoRL 2020

Meta-SAC. Auto-tune the Entropy Temperature of Soft Actor-Critic via Metagradient - 7th ICML AutoML workshop 2020

self-predictive-rl. Bridging State and History Representations: Understanding Self-Predictive RL, ICLR 2024

llm-reasoning-uft. Code for Offline Learning and Forgetting for Reasoning with Large Language Models, TMLR 2025

neubay. JAX code for "Long-Horizon Model-Based Offline Reinforcement Learning Without Explicit Conservatism", ICML 2026

Anonymous-Forum. PKU Network Practice