Trax — Deep Learning with Clear Code and Speed
https://github.com/google/trax
688 forks.
6,721 stars.
87 open issues.
Recent commits:
- Fix test failures caused by a NumPy 1.21 upgradePiperOrigin-RevId: 418075062, Copybara-Service
- Fix a failing initializer test.PiperOrigin-RevId: 414013921, Copybara-Service
- Fixes testTensordot flakiness by relaxing tolerance.PiperOrigin-RevId: 410351568, Copybara-Service
- [JAX] Update users of `jax.ops.index…` functions, which are deprecated.* replace uses of `jax.ops.index[…]` with `jax.numpy.index_exp[…]`, which is a standard NumPy function that does the same thing.* remove some redundant uses of `jax.ops.index[…]`, where the expression is passed directly to an indexed accessor function like `.at[…]`.* update some remaining users of `jax.ops.index_update(x, jax.ops.index[idx], y)` to use the `x.at[idx].set(y)` APIs.PiperOrigin-RevId: 405825513, Copybara-Service
- TRAX v.1.4.1 and updates to the release script for python3.PiperOrigin-RevId: 405721113, Copybara-Service
Was this post helpful?
Let us know if you liked the post. That’s the only way we can improve.