All pull requests

Pull requests

• fix(rvt): AxialRotaryEmbedding preserves float16 / bfloat16 input dtype

  #373·tritsystem opened Sep 4, 2026·

  ·

• Guard conv_output_size against a zero stride; use floor division

  #367·CharlesCNorton opened Jun 23, 2026·

  ·

• Missing and inconsistent einsum operations

  #363·tahashieenavaz opened Apr 28, 2026·

  ·
  1

• Update README.md

  #343·saharnazyp opened Aug 2, 2025·

  ·

• Modified the forward function of Attention so that it can operate on …

  #334·EdgeObserver opened Nov 15, 2024·

  ·

• Update distill.py to include device agnostic code for distill_mlp head and distillation_token

  #324·vivekh2000 opened Jul 25, 2024·

  ·

• Update simple_flash_attn_vit_3d.py

  #323·zhulinchng opened Jul 9, 2024·

  ·

• Update distill.py

  #318·vivekh2000 opened Jun 11, 2024·

  ·

• Update vit.py to include LayerNorm in the MLP head which is missing

  #317·vivekh2000 opened Jun 11, 2024·

  ·
  1

• display random images when previewing in cats_and_dogs.ipynb

  #301·berinaniesh opened Mar 25, 2024·

  ·

• First attempt

  #290·jefferson-bercaw opened Nov 28, 2023·

  ·

• Some questions about decoder position embedding for masked tokens

  #173·chrisway613 opened Nov 24, 2021·

  ·
  1

• How to save and restore the parameters?

  #162·lesept777 opened Oct 6, 2021·

  ·