All pull requests

Pull requests

• Close interleaved prefetch workers concurrently

  #1430·jabrailkhalil opened Oct 12, 2026·

  ·
  1

• Drop dead iterator refs from the stats registry

  #1429·Shr3yash opened Oct 12, 2026·

  ·
  1

• Raise output pickling errors from process prefetch workers

  #1428·rioyu123 opened Oct 10, 2026·

  ·
  1

• Fixed possible cleanup failures in SharedMemoryArray.

  #1422·copybara-service[bot] opened Oct 8, 2026Bot·

  ·

• Fix shared memory closed by two threads after mp_prefetch finishes a shard

  #1421·mahdi-shafiei opened Oct 8, 2026·

  ·
  1

• Enable batched read pushdown (SupportsBatchedReadRandomAccessDataSource) on grain.python.ArrayRecordDataSource.

  #1412·copybara-service[bot] opened Oct 4, 2026Bot·

  ·

• Internal

  #1389·copybara-service[bot] opened Sep 2, 2026Bot·

  ·

• Fix SharedMemoryArray.__del__ AttributeError during interpreter shutdown

  #1386·rootkiller6788 opened Aug 20, 2026·

  ·

• internal

  #1383·copybara-service[bot] opened Aug 7, 2026Bot·

  ·

• Internal

  #1382·copybara-service[bot] opened Aug 7, 2026Bot·

  ·

• internal

  #1380·copybara-service[bot] opened Aug 6, 2026Bot·

  ·

• Internal

  #1379·copybara-service[bot] opened Aug 6, 2026Bot·

  ·

• Internal

  #1378·copybara-service[bot] opened Aug 6, 2026Bot·

  ·

• Internal

  #1377·copybara-service[bot] opened Aug 6, 2026Bot·

  ·

• Add BagzDataSource for BagZ file format support

  #1376·Marlon666 opened Aug 4, 2026·

  ·

• internal

  #1374·copybara-service[bot] opened Aug 3, 2026Bot·

  ·

• Fix ThreadPrefetch close deadlock when producer blocks in parent

  #1373·MohammadSadeghSalehi opened Aug 2, 2026·

  ·
  1

  3

• Fix MultiprocessingOptions.enable_profiling being ignored.

  #1370·copybara-service[bot] opened Jul 30, 2026Bot·

  ·

• Track read storage type metric on checkpoint load and restore operations.

  #1339·copybara-service[bot] opened Jun 6, 2026Bot·

  ·

• Allow exporting PyGrain telemetry metrics to open-source Prometheus.

  #1338·copybara-service[bot] opened Jun 6, 2026Bot·

  ·

• Enable lazy imports for stats and stats_utils targets to improve startup time.

  #1331·copybara-service[bot] opened Jun 2, 2026Bot·

  ·

• Update TfGrainMitxure to provide batch size to TfMixtureIndexSampler

  #1327·copybara-service[bot] opened May 20, 2026Bot·

  ·

• Skip scaling proportions if the minimun no-zero value equal to scale_min_to in IndexSampler

  #1314·copybara-service[bot] opened May 19, 2026Bot·

  ·

• improve ParquetIterDataset docstring and tutorial docs

  #1298·copybara-service[bot] opened May 5, 2026Bot·

  ·
  1

• Internal

  #1296·copybara-service[bot] opened May 2, 2026Bot·

  ·