This repository provides the official implementation for our paper: Steering Your Diffusion Policy with Latent Space Reinforcement Learning (CoRL 2025). Specifically, it contains a JAX-based ...
Abstract: A smart city is an urban development idea that seeks to efficiently incorporate diverse information and communication technology (ICT) solutions in a safe manner to properly oversee a city’s ...
flash-attention-with-sink implements an attention variant used in GPT-OSS 20B that integrates a "sink" step into FlashAttention. This repo focuses on the forward path and provides an experimental ...
Abstract: In this study, we propose LWMalloc, a lightweight dynamic memory allocator designed for resource-constrained environments. LWMalloc incorporates a lightweight data structure, a deferred ...