All pull requests

Pull requests

• fix(torch): make MinMaxRandomDataset iterable so inspect() can use it

  review-effort/mediumtests
  #156·v0ropaev opened Oct 31, 2026Contributor·

  ·
  6

• fix(torch): honor config.cache_dir for AOT module caches and clear_cache

  review-effort/lowteststuning
  #155·v0ropaev opened Oct 30, 2026Contributor·

  ·
  6

• fix(checkpoint): omit original weights from JIT checkpoints

  backendsdocumentationreview-effort/mediumtests
  #154·knowicki-nvidia opened Oct 30, 2026Collaborator·

  ·
  14

  

• fix(jit): restore module hooks when unpatching inspected modules

  jitreview-effort/lowtests
  #153·jakkisel opened Oct 30, 2026Contributor·

  ·
  4

  

• test(functional_ci): simplify toy Flux pipeline and run it on all backends

  review-effort/mediumtests
  #152·knowicki-nvidia opened Oct 30, 2026Collaborator·

  ·
  9

  

• fix(backends): require and preserve SampleStore for checkpoint warmup

  backendsreview-effort/hightests
  #151·jkosek opened Oct 30, 2026Contributor·

  ·
  9

  

• fix(torch): preserve keyword replay for NeMo modules

  documentationtests
  #144·glos-nv opened Oct 28, 2026·

  ·
  2

• test(torch): fix typing errors and prune baseline

  review-effort/lowtests
  #143·glos-nv opened Oct 28, 2026·

  ·
  1

• refactor(wrapping): drop root cast in wrap with an invariant assert

  inspectionreview-effort/low
  #142·jpohl-nv opened Oct 28, 2026Contributor·3/3·

  ·
  3

  

• chore(typing): make TorchCheckpoint.load generic and drop cast in ait.load

  checkpointreview-effort/lowtuning
  #140·jpohl-nv opened Oct 28, 2026Contributor·2/3·

  ·
  2

  

• test(functional-ci): cover multiple graphs on all backends for AOT and JIT tuning

  review-effort/mediumtests
  #139·knowicki-nvidia opened Oct 28, 2026Collaborator·

  ·
  15

  

• test(functional_ci): cover static shapes and save/load on JIT backends

  backendsreview-effort/mediumtests
  #138·jpohl-nv opened Oct 27, 2026Contributor·

  ·
  24

  

• fix(kernel_forge): tear down cupti after kernel profiling

  bugprofilingreview-effort/mediumtests
  #135·szysad opened Oct 27, 2026Contributor·

  ·
  24

  

• feat(jit): integrate FlexTensor-managed offload

  dependenciesdocumentationexamplesjitreview-effort/hightests
  #89·kacper-kleczewski opened Oct 15, 2026Contributor·3/3·

  ·
  18

• examples: flux schnell

  dependenciesdocumentationdynamoexamplesreview-effort/high
  #85·glos-nv opened Oct 15, 2026·

  ·
  39

  

• feat(inductor): add export hook to expose the FX graph before AOT compile

  backendsdocumentationreview-effort/hightests
  #81·knowicki-nvidia opened Oct 13, 2026Collaborator·

  ·
  13

  

• feat(jit): support externally managed module placement

  backendsdocumentationjitreview-effort/highteststuning
  #45·kacper-kleczewski opened Sep 28, 2026Contributor·2/3·

  ·
  24

  

• feat(jit): add explicit module registration

  cidocumentationjitreview-effort/hightests
  #43·kacper-kleczewski opened Sep 28, 2026Contributor·1/3·

  ·
  32