google trax issues - Githubissues

google / trax

Trax — Deep Learning with Clear Code and Speed

Apache License 2.0

8.01k stars 813 forks source link

issues

Newest

Newest Most commented Recently updated Oldest Least commented Least recently updated

Replace deprecated `jax.tree_*` functions with `jax.tree.*`

#1797 copybara-service[bot] closed 1 month ago
0
Remove uses of deprecated jax.lax.tie_in

#1796 copybara-service[bot] closed 5 months ago
0
Remove uses of deprecated jax.lax.tie_in

#1795 copybara-service[bot] opened 5 months ago
0
Inconsistency in function's doc-string

#1794 jaindhairyahere opened 5 months ago
1
Replace references to deprecated device_buffer attributes

#1793 copybara-service[bot] opened 7 months ago
0
Refactoring to 1.5.1

#1792 mmarcinmichal opened 7 months ago
1
[JAX] Replace uses of jax.devices("cpu") with jax.local_devices(backend="cpu").

#1791 copybara-service[bot] closed 7 months ago
0
Error loading loop from a checkpoint

#1790 nwbvt opened 8 months ago
1
trax: avoid referencing deprecated jax.random.threefry_2x32

#1789 copybara-service[bot] closed 8 months ago
0
Updated jax.config import

#1788 copybara-service[bot] closed 8 months ago
0
AttributeError: 'function' object has no attribute 'n_steps_per_checkpoint' for NLP Machine translation model

#1787 Sumit1673 opened 9 months ago
1
Are any easy ways to use something like `train_test_split` from `sklearn`?

#1786 mihalt opened 9 months ago
0
Can't run `bert_vocab_from_dataset` without `TypeError: Tensor is unhashable` when import `trax` with `tensorflow`

#1785 mihalt opened 9 months ago
0
Update core.py: fix docstring "y = Wx + b" -> "y = xW + b"

#1784 arvyzukai opened 10 months ago
1
Can I do simple tokenization?

#1783 mihalt opened 10 months ago
0
fix: write file typo

#1782 0o001 opened 10 months ago
0
Nick dory patch 1

#1781 NickDory opened 10 months ago
1
Is possible Linformer algorithm ?

#1780 andr1972 opened 11 months ago
0
Issue when running training_loop.run(2000) - message StopIteration in next_batch(self)

#1779 SisypheGeek opened 1 year ago
0
Could not normally run trax using GPU in local computer

#1778 LiuZhenshun opened 1 year ago
0
Move the TF-NumPy einsum test from Trax to the TF-NumPy directory.

#1777 copybara-service[bot] opened 1 year ago
0
Cannot import Trax

#1776 Nuna7 opened 1 year ago
6
Remove references to deprecated jax.ShapedArray

#1775 copybara-service[bot] closed 1 year ago
0
Unable to import trax

#1774 JoseJamesMD opened 1 year ago
1
AttributeError: module 'jax.ops' has no attribute 'index_add'

#1773 cmosguy opened 1 year ago
1
[JAX] Replace uses of jax.xla_computation() with jax.jit().lower().

#1772 copybara-service[bot] closed 1 year ago
0
SelfAttention - problem with tensorflow 2.11.0

#1771 mmarcinmichal opened 1 year ago
0
TypeError: unsupported operand type(s) for ==: 'Array' and 'tuple'

#1770 mbateman opened 1 year ago
2
Limit the dataset from TFDS

#1769 oendnsk675 opened 1 year ago
0
how to use trax to translate other languages

#1768 SumanthSrungavarapu opened 1 year ago
0
[NumPy] Remove references to deprecated NumPy type aliases.

#1767 copybara-service[bot] closed 1 year ago
0
ImportError: cannot import name 'MergeHeads' from 'trax.layers.attention'

#1766 jpontalba opened 1 year ago
0
Machine Translation Refromer model.pkl for trax 1.4.1?

#1765 ymcki opened 1 year ago
0
fix: broken "Open in Colab" link in Knowledge Tracing Notebook

#1764 SauravMaheshkar closed 1 year ago
0
Disables a failing remat test.

#1763 copybara-service[bot] opened 1 year ago
0
[tf.numpy] Fixes jax_tests/ breakages.

#1762 copybara-service[bot] closed 1 year ago
0
Update inputs.py

#1761 chrisrobertsoncypress opened 1 year ago
1
TypeError: float() argument must be a string or a number, not 'jaxlib.tpu_client_extension.PyTpuBuffer'

#1760 agoliaei opened 1 year ago
0
Make this code compatible with Python 3.10.

#1759 copybara-service[bot] closed 1 year ago
0
The colab button on Knowledge_Tracing_Transformer.ipynb is not open

#1758 haytamdon opened 1 year ago
0
[NumPy] Replace numpy.asscalar(x) with x.item() in preparation for upgrading NumPy to 1.23.

#1757 copybara-service[bot] closed 1 year ago
0
A change to type checking in checkpoint loading.

#1756 copybara-service[bot] closed 1 year ago
0
[NumPy] Fix uses of deprecated multidimensional NumPy indexing with a non-tuple index.

#1755 copybara-service[bot] closed 1 year ago
0
fix: broken link in `trax.data` example notebook

#1754 SauravMaheshkar closed 1 year ago
0
TypeError: external_configurable() got an unexpected keyword argument 'blacklist'

#1753 NirajanBekoju opened 1 year ago
0
Not able to import trax in google colab though trax is installed

#1752 ankusri opened 2 years ago
1
Disable broken trax tests

#1751 copybara-service[bot] closed 2 years ago
0
Please look carefully at this CL.

#1750 copybara-service[bot] closed 2 years ago
0
Does the Reformer have more parameters than the baseline?

#1749 alexm-gc opened 2 years ago
0
Does the Reformer have more parameters than the baseline?

#1748 alexm-gc closed 2 years ago
1