google/trax

Issues

• Cannot import Trax

  #1776 opened 2 years ago by Nuna7
  8

• Inconsistency in function's doc-string

  #1794 opened a year ago by jaindhairyahere
  1

• Error loading loop from a checkpoint

  #1790 opened a year ago by nwbvt
  1

• ValueError: no signature found for builtin function

  #1730 opened 3 years ago by alistairmayo
  4

• AttributeError: 'function' object has no attribute 'n_steps_per_checkpoint' for NLP Machine translation model

  #1787 opened 2 years ago by Sumit1673
  1

• Are any easy ways to use something like `train_test_split` from `sklearn`?

  #1786 opened 2 years ago by mihalt
  0

• Can't run `bert_vocab_from_dataset` without `TypeError: Tensor is unhashable` when import `trax` with `tensorflow`

  #1785 opened 2 years ago by mihalt
  0

• Can I do simple tokenization?

  #1783 opened 2 years ago by mihalt
  0

• Is possible Linformer algorithm ?

  #1780 opened 2 years ago by andr1972
  0

• Issue when running training_loop.run(2000) - message StopIteration in next_batch(self)

  #1779 opened 2 years ago by SisypheGeek
  0

• Could not normally run trax using GPU in local computer

  #1778 opened 2 years ago by LiuZhenshun
  0

• TypeError: unsupported operand type(s) for ==: 'Array' and 'tuple'

  #1770 opened 2 years ago by mbateman
  2

• AttributeError: module 'jax.ops' has no attribute 'index_add'

  #1773 opened 2 years ago by cmosguy
  1

• Unable to import trax

  #1774 opened 2 years ago by JoseJamesMD
  1

• SelfAttention - problem with tensorflow 2.11.0

  #1771 opened 2 years ago by mmarcinmichal
  0

• Limit the dataset from TFDS

  #1769 opened 2 years ago by oendnsk675
  0

• how to use trax to translate other languages

  #1768 opened 2 years ago by SumanthSrungavarapu
  0

• Not able to import trax in google colab though trax is installed

  #1752 opened 3 years ago by ankusri
  1

• ImportError: cannot import name 'MergeHeads' from 'trax.layers.attention'

  #1766 opened 2 years ago by jpontalba
  0

• Trax ML: GPU memory allocated but completed on CPU

  #1675 opened 4 years ago
  5

• Machine Translation Refromer model.pkl for trax 1.4.1?

  #1765 opened 2 years ago by ymcki
  0

• TypeError: float() argument must be a string or a number, not 'jaxlib.tpu_client_extension.PyTpuBuffer'

  #1760 opened 3 years ago by agoliaei
  0

• The colab button on Knowledge_Tracing_Transformer.ipynb is not open

  #1758 opened 3 years ago by haytamdon
  0

• Trax.data API explained -> Open in Colab errors

  #1744 opened 3 years ago by ras44
  1

• TypeError: external_configurable() got an unexpected keyword argument 'blacklist'

  #1753 opened 3 years ago by NirajanBekoju
  0

• Does the Reformer have more parameters than the baseline?

  #1749 opened 3 years ago by alexm-gc
  0

• Does the Reformer have more parameters than the baseline?

  #1748 opened 3 years ago by alexm-gc
  1

• Do we have a reading tasking for question and answering?

  #1732 opened 3 years ago by AliceSum
  1

• Model output various predicted result without re-init-model

  #1731 opened 3 years ago by kizcas
  0

• Can Trax be used for Arabic NMT?

  #1728 opened 3 years ago by AliMi001
  0

• How to pass data carried by a class through forward pass

  #1719 opened 3 years ago by aycandv
  0

• Cannot Train LSHSelfAttention on GPU

  #1724 opened 3 years ago by jsearcy1
  0

• Trax Reserves All Available GPU Memory

  #1722 opened 3 years ago by rafidka
  2

• unmapped_aval() missing 1 required positional argument: 'aval'

  #1720 opened 3 years ago by moeenm
  1

• Cannot export model from trax to TF

  #1718 opened 3 years ago by AndriCcos
  1

• nan CrossEntropyLossWithLogSoftmax while training an NMT Reformer

  #1717 opened 3 years ago by amrnablus
  0

• AttributeError: module 'trax.supervised.trainer_lib' has no attribute 'init_random_number_generators'

  #1714 opened 3 years ago by topekekere
  1

• ValueError on predict mode Transformer model

  #1670 opened 4 years ago by shashank2123
  2

• RAM crash while optimizing a linear regression model

  #1697 opened 4 years ago by aycandv
  0

• Relu layer nested in its own Serial combinator by default

  #1695 opened 4 years ago by kbvw
  1

• Multiple heads option is not working in SelfAttention

  #1686 opened 4 years ago by kenenbek
  1

• Wrapping up a combinator layer

  #1681 opened 4 years ago by aycandv
  2

• Reformer does not export to SavedModel format.

  #1685 opened 4 years ago by StewartSethA
  0

• How to make predictions on a trained model

  #1682 opened 4 years ago by TurquoiseA
  0

• L2 & L1 Regularizers in Trax

  #1669 opened 4 years ago
  0

• LSHSelfAttention only works with different lengths if the parameter use_reference_code is set to "True"

  #1664 opened 4 years ago by renevs
  0

• tl.Relu and tl.ShiftRight layers gets inside Serial Combinator

  #1660 opened 4 years ago by apoorvgupta25
  1

• Trax Import Error

  #1661 opened 4 years ago by memora0101
  0

• BERT model fails on its initialization

  #1655 opened 4 years ago by manifest
  2

• Trax reformer description

  #1654 opened 4 years ago by kenenbek
  0