Issues

• stats._iter_weakref_registry keeps refs of dead iterators, and a new iterator at a reused address is not registered

  type:bug
  google/grain#1427 · Filter by author mahdi-shafiei opened on Oct 10, 2026

• mp_prefetch worker processes keep running after the main process is killed (SIGKILL/SIGTERM)

  type:bug
  google/grain#1423 · Filter by author mahdi-shafiei opened on Oct 8, 2026

• mp_prefetch: a reference cycle per finished shard, and SharedMemory closed by two threads

  type:bug
  google/grain#1420 · Filter by author mahdi-shafiei opened on Oct 8, 2026

• mp_prefetch silently drops elements that cannot be pickled

  type:bug
  google/grain#1415 · Filter by author mahdi-shafiei opened on Oct 6, 2026

• Release request: include a3b351858a (unparsed worker profiling flag); grain-nightly has not published since 2026-01-12

  type:bug
  google/grain#1411 · Filter by author mahdi-shafiei opened on Oct 3, 2026
  1

• Closing an mp_prefetch iterator stops its worker processes one at a time

  type:bug
  google/grain#1410 · Filter by author AshishKumar4 opened on Oct 1, 2026

• ThreadPrefetch close() cannot cancel a producer blocked in parent.__next__()

  type:bug
  google/grain#1372 · Filter by author MohammadSadeghSalehi opened on Aug 2, 2026

• How are config options meant to be set in Grain?

  type:support
  google/grain#1360 · Filter by author anihm136 opened on Jul 15, 2026

• Sliding window operation in IterDataset / build grain locally

  type:support
  google/grain#1264 · Filter by author MathiesW opened on Mar 26, 2026
  1

• No support for python 3.13t and 3.14t

  type:feature
  google/grain#1245 · Filter by author aurelio-amerio opened on Mar 17, 2026
  1

• Issue on page /grain.experimental.html

  documentation
  google/grain#1238 · Filter by author AakashKumarNain opened on Mar 7, 2026
  4

• Add Windows Build to conda-forge

  type:feature
  google/grain#1232 · Filter by author RoyiAvital opened on Feb 27, 2026

• Grain workers use GPUs, and there is no way to avoid it

  type:performance
  google/grain#1199 · Filter by author AakashKumarNain opened on Jan 17, 2026
  5

• ThreadPrefetchIterDataset memory leaks / zombie threads that hang exiting?

  type:bug
  google/grain#1196 · Filter by author am001122 opened on Jan 15, 2026
  3

• Performance gap with tf.data.Dataset on MNIST

  type:performance
  google/grain#1164 · Filter by author rademacher-p opened on Dec 15, 2025
  5

• random and augmentation in datasets

  type:support
  google/grain#1102 · Filter by author pcasl opened on Oct 29, 2025
  1

• grain dataset slow with huggingface data source

  type:performance
  google/grain#1084 · Filter by author aurelio-amerio opened on Oct 16, 2025

• Leaked shared_memory objects to clean up at shutdown

  type:bug
  google/grain#1031 · Filter by author DBraun opened on Sep 3, 2025
  1

• MapDataset does not respect jax.default_device(jax.devices("cpu")[0]).

  type:bug
  google/grain#975 · Filter by author james77777778 opened on Aug 6, 2025
  3

• Add dataloader checkpointing example to docs

  documentationtype:feature
  google/grain#938 · Filter by author emergenz opened on Jul 10, 2025
  2

• DataLoader ignores context manager for Data Sources

  type:support
  google/grain#936 · Filter by author jimtjames opened on Jul 5, 2025

• Document backwards compatibility guarantees

  type:feature
  google/grain#898 · Filter by author kerrickstaley opened on Jun 5, 2025
  1

• Feature: Define transforms via Protocol instead of abstract class

  type:feature
  google/grain#884 · Filter by author JEM-Mosig opened on May 28, 2025
  2

• Discussion: Grain packaging nightly builds

  type:build/install
  google/grain#878 · Filter by author mtsokol opened on May 21, 2025
  2

• SharedMemoryArray is too slow

  type:performance
  google/grain#851 · Filter by author tridemax opened on May 5, 2025
  3