All pull requests

Pull requests

• Fix cross-replica masked batch normalization

  #1243·sylvesterkaczmarek opened Sep 10, 2026·

  ·

• [SAM2] Vectorize backbone image encoder by flattening (B, T) instead of jax.vmap.

  #1242·copybara-service[bot] opened Aug 28, 2026Bot·

  ·

• Bump pillow from 9.1.1 to 12.3.0 in /scenic/projects/ncr

  dependenciespython
  #1236·dependabot[bot] opened Jul 22, 2026Bot·

  ·

• Use jnp.zeros rather than jnp.empty in places where values are important.

  #1209·copybara-service[bot] opened Apr 15, 2026Bot·

  ·

• Internal

  #1204·copybara-service[bot] opened Mar 5, 2026Bot·

  ·

• chore: Migrate gsutil usage to gcloud storage

  #1203·risha-vijayvargiya-22 opened Mar 3, 2026·

  ·

• Update jax.host_id() and jax.host_count() to jax.process_index() and jax.process_count(), respectively.

  #1194·copybara-service[bot] opened Feb 4, 2026Bot·

  ·

• Feature/efficient attention

  #1189·divye-joshi opened Jan 27, 2026·

  ·

• No public description

  #1185·copybara-service[bot] opened Dec 19, 2025Bot·

  ·
  1

• No public description

  #1184·copybara-service[bot] opened Dec 19, 2025Bot·

  ·
  1

• local testing only

  #1182·copybara-service[bot] opened Dec 9, 2025Bot·

  ·
  1

• Small refactor and cleanup of trainer.py.

  #1161·copybara-service[bot] opened Apr 28, 2025Bot·

  ·
  1

• Added up down flip to the scenic input pipeline.

  #1155·copybara-service[bot] opened Apr 8, 2025Bot·

  ·
  1

• Internal

  #1151·copybara-service[bot] opened Feb 25, 2025Bot·

  ·

• Enable loading ImageNet file_name field to support loading ImageNet captions by file_name

  #1141·copybara-service[bot] opened Dec 29, 2024Bot·

  ·
  1

• Internal.

  #1134·copybara-service[bot] opened Nov 27, 2024Bot·

  ·

• ...cleanup and resolve pytype issues with assertions...

  #1130·copybara-service[bot] opened Nov 21, 2024Bot·

  ·
  1

• Ignore some pytype errors caused by an internal Pillow upgrade.

  #1114·copybara-service[bot] opened Oct 2, 2024Bot·

  ·

• Ignore some pytype errors caused by an internal Pillow upgrade.

  #1113·copybara-service[bot] opened Oct 2, 2024Bot·

  ·

• First entry for Draft (Directly fine-tuning diffusion models on differentiable rewards) codebase.

  #1078·copybara-service[bot] opened Jun 21, 2024Bot·

  ·

• internal

  #1070·copybara-service[bot] opened May 29, 2024Bot·

  ·

• Update input_pipeline.py

  #1069·TheMattBin opened May 26, 2024·

  ·

• Commented out lines of the verbs_in_action project for code cleanup

  #1067·Mas-Ayb opened May 22, 2024·

  ·
  1

• Refactored DETR implementation on optax API.

  #1062·MasterSkepticista opened May 13, 2024·

  ·
  2

• No change.

  #1057·copybara-service[bot] opened May 7, 2024Bot·

  ·
  1