Skip to content

@heavyball-research HeavyBall Research

All

    Repositories list

    8 repositories

    • heavyball-research.github.io

      Public
      CSS
      •0•0•0•0•Updated Oct 17, 2026Oct 17, 2026

    • flywheel-tpu

      Public
      Fast, open-source Pallas kernels for TPU inference and training.
      Python
      •
      Apache License 2.0
      •3•11•1•2•Updated Oct 17, 2026Oct 17, 2026

    • tpu-inference

      Public
      TPU inference for vLLM, with unified JAX and PyTorch support.
      Python
      •
      Apache License 2.0
      •345•0•0•1•Updated Oct 16, 2026Oct 16, 2026

    • .github

      Public
      0•0•0•0•Updated Sep 29, 2026Sep 29, 2026

    • multi-token-residual-prediction

      Public
      A lightweight draft module that learns to predict several future denoising steps from a single target-model forward pass, enabling speculative-style decoding th…
      Python
      •
      MIT License
      •0•9•0•0•Updated Sep 7, 2026Sep 7, 2026

    • computing-resources-guide

      Public
      A comprehensive collection of guidelines to all the computing resources the Heavyball Lab @ NYU Shanghai can access.
      Python
      •0•5•0•0•Updated Jun 24, 2026Jun 24, 2026

    • sglang

      Public
      Python
      •
      Apache License 2.0
      •0•0•0•0•Updated May 16, 2026May 16, 2026

    • DistillBytes

      Public
      Two-Stage Framework Distilling Token-Trained Models into Byte-Level Models
      Python
      •
      MIT License
      •0•1•0•0•Updated Feb 11, 2026Feb 11, 2026
    ProTip! When viewing an organization's repositories, you can use the props. filter to filter by custom property.