issues
search
google
/
trax
Trax — Deep Learning with Clear Code and Speed
Apache License 2.0
8.07k
stars
813
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
disable certain tests in lax_numpy_test when numpy 2.0 is used
#1804
copybara-service[bot]
closed
3 weeks ago
0
disable testOp when numpy 2.0 is installed
#1803
copybara-service[bot]
closed
3 weeks ago
0
fix a few test bitwise op tests for np 2.0
#1802
copybara-service[bot]
closed
3 weeks ago
0
skip testLinSpace and testLogSpace if numpy 2.0 is used
#1801
copybara-service[bot]
closed
1 month ago
0
[numpy] Fix test failures under NumPy 2.0.
#1800
copybara-service[bot]
closed
1 month ago
0
[numpy] Fix test failures under NumPy 2.0.
#1799
copybara-service[bot]
closed
1 month ago
0
[scipy] Use NumPy functions rather than their deprecated aliases in SciPy's namespace.
#1798
copybara-service[bot]
closed
2 months ago
0
Replace deprecated `jax.tree_*` functions with `jax.tree.*`
#1797
copybara-service[bot]
closed
4 months ago
0
Remove uses of deprecated jax.lax.tie_in
#1796
copybara-service[bot]
closed
8 months ago
0
Remove uses of deprecated jax.lax.tie_in
#1795
copybara-service[bot]
opened
8 months ago
0
Inconsistency in function's doc-string
#1794
jaindhairyahere
opened
8 months ago
1
Replace references to deprecated device_buffer attributes
#1793
copybara-service[bot]
opened
10 months ago
0
Refactoring to 1.5.1
#1792
mmarcinmichal
opened
10 months ago
1
[JAX] Replace uses of jax.devices("cpu") with jax.local_devices(backend="cpu").
#1791
copybara-service[bot]
closed
10 months ago
0
Error loading loop from a checkpoint
#1790
nwbvt
opened
11 months ago
1
trax: avoid referencing deprecated jax.random.threefry_2x32
#1789
copybara-service[bot]
closed
11 months ago
0
Updated jax.config import
#1788
copybara-service[bot]
closed
11 months ago
0
AttributeError: 'function' object has no attribute 'n_steps_per_checkpoint' for NLP Machine translation model
#1787
Sumit1673
opened
1 year ago
1
Are any easy ways to use something like `train_test_split` from `sklearn`?
#1786
mihalt
opened
1 year ago
0
Can't run `bert_vocab_from_dataset` without `TypeError: Tensor is unhashable` when import `trax` with `tensorflow`
#1785
mihalt
opened
1 year ago
0
Update core.py: fix docstring "y = Wx + b" -> "y = xW + b"
#1784
arvyzukai
opened
1 year ago
1
Can I do simple tokenization?
#1783
mihalt
opened
1 year ago
0
fix: write file typo
#1782
0o001
opened
1 year ago
0
Nick dory patch 1
#1781
NickDory
opened
1 year ago
1
Is possible Linformer algorithm ?
#1780
andr1972
opened
1 year 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
8
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
2 years ago
1
TypeError: float() argument must be a string or a number, not 'jaxlib.tpu_client_extension.PyTpuBuffer'
#1760
agoliaei
opened
2 years ago
0
Make this code compatible with Python 3.10.
#1759
copybara-service[bot]
closed
2 years ago
0
The colab button on Knowledge_Tracing_Transformer.ipynb is not open
#1758
haytamdon
opened
2 years ago
0
[NumPy] Replace numpy.asscalar(x) with x.item() in preparation for upgrading NumPy to 1.23.
#1757
copybara-service[bot]
closed
2 years ago
0
A change to type checking in checkpoint loading.
#1756
copybara-service[bot]
closed
2 years ago
0
[NumPy] Fix uses of deprecated multidimensional NumPy indexing with a non-tuple index.
#1755
copybara-service[bot]
closed
2 years ago
0
Next