deepmind/jmp issues and pull requests

#46 - Internal change

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#45 - Example on JMP x Flax?

Issue - State: closed - Opened by llCurious over 1 year ago - 2 comments

#44 - Update loss_scale.py

Pull Request - State: open - Opened by cyprienc over 1 year ago

#43 - Fix or ignore some pytype errors related to jnp.ndarray == jax.Array.

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago

#42 - Bump version to 0.0.5.dev.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#41 - Cut 0.0.4 release.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#40 - Remove two warnings that were showing when using jmp.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#39 - Warnings during runtime

Issue - State: closed - Opened by psmaragdis almost 2 years ago - 1 comment

#38 - Bump JMP version to 0.0.4.dev.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#37 - Cut 0.0.3 release.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#36 - Drop Python 3.7 support.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#35 - Copybara import of the project:

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#34 - New release?

Issue - State: closed - Opened by mariogeiger almost 2 years ago - 2 comments

#33 - Add absl-py test dep to fix github actions test.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#32 - CI also with python 3.10 and 3.11

Pull Request - State: closed - Opened by mariogeiger almost 2 years ago - 2 comments

#31 - Add gitignore

Pull Request - State: closed - Opened by NeilGirdhar almost 2 years ago

#30 - Fix dataclass use failure on Python 3.11

Pull Request - State: closed - Opened by NeilGirdhar almost 2 years ago

#29 - Overload str convenience method to provide str precision specs from jmp.Policy.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago - 1 comment

#28 - Questions around speedup

Issue - State: closed - Opened by 1m1ne about 2 years ago - 3 comments

#27 - Support cast_to_${x} on numpy arrays.

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#26 - Add documentation that cast_to_$x only casts floating values.

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#25 - Replace jax.lax.select with jnp.where

Pull Request - State: closed - Opened by nlsfnr about 2 years ago - 6 comments

#24 - Use jax.tree_util.tree_map in place of deprecated tree_multimap.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago

#23 - Changing jax.tree_map -> jax.tree_util.tree_map due to recent deprecation.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago - 1 comment

#22 - Basic question about the use of my_policy

Issue - State: closed - Opened by JSchuurmans over 2 years ago - 2 comments

#21 - Enable configurable minimum loss scale.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago
Labels: cla: yes

#20 - Ensure requirements*.txt are included in source distribution.

Pull Request - State: closed - Opened by copybara-service[bot] over 2 years ago
Labels: cla: yes

#19 - jmp-0.0.2.tar.gz on PyPY doesn't contain requirements.txt

Issue - State: closed - Opened by ostueker over 2 years ago - 1 comment

#18 - Configure minimum dynamic loss scale

Issue - State: closed - Opened by danijar over 2 years ago - 1 comment

#17 - Casting numpy array

Issue - State: closed - Opened by yardenas almost 3 years ago - 1 comment

#16 - Remove build file comment for dataclasses.

Pull Request - State: closed - Opened by copybara-service[bot] about 3 years ago
Labels: cla: yes

#15 - Update setup.py

Pull Request - State: closed - Opened by anukaal about 3 years ago
Labels: cla: yes

#14 - Stop checking Python 3.6, no longer supported by JAX.

Pull Request - State: closed - Opened by copybara-service[bot] about 3 years ago
Labels: cla: yes

#13 - Parse JAX extra packages from requirements-jax.txt.

Pull Request - State: closed - Opened by copybara-service[bot] about 3 years ago
Labels: cla: yes

#12 - Install the latest version of JAX.

Pull Request - State: closed - Opened by copybara-service[bot] about 3 years ago
Labels: cla: yes

#11 - Add support for internal usage logging.

Pull Request - State: closed - Opened by copybara-service[bot] about 3 years ago
Labels: cla: yes