Skip to content
@heavyball-research

HeavyBall Research

ML System Lab @ NYU Shanghai

Pinned Loading

  1. flywheel-tpu flywheel-tpu Public

    Fast, open-source Pallas kernels for TPU inference and training.

    Python 9 1

  2. multi-token-residual-prediction 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 that yields the same outputs with subst…

    Python 9

Repositories

Showing 7 of 7 repositories

People

This organization has no public members. You must be a member to see who’s a part of this organization.

Top languages

Loading…

Most used topics

Loading…