mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-21 22:36:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
5,166 Commits 508 Branches 263 Tags
Commit Graph

42 Commits

Author SHA1 Message Date
Peter Hawkins
f036f5ddb0
Avoid direct type/dtype comparisons to fix NumPy 1.19 deprecation war… (#3543) 
* Avoid direct type/dtype comparisons to fix NumPy 1.19 deprecation warnings.

* Pin a newer tf-nightly to fix jax2tf tests for NumPy 1.19.0
 2020-06-24 15:19:00 -04:00
Matthew Johnson
3fff837907
pin numpy version in setup.py to avoid warnings (#3509) 2020-06-22 08:12:41 -07:00
Peter Hawkins
cbdf9a5a43
Drop support for Python 3.5. (#2445) 2020-03-18 10:54:28 -04:00
Peter Hawkins
681ba37f7e
Drop fastcache dependency, which isn't necessary on Python 3. (#1995) 
Drop protobuf and six dependencies from travis configuration.
 2020-01-14 10:08:23 -05:00
Peter Hawkins
dcc882cf6b
Drop Python 2 support from JAX. (#1962) 
Remove six dependency.
 2020-01-08 13:17:55 -05:00
Peter Hawkins
4fc765241f
Drop protobuf dependency from jax package. It appears unused. (#1700) 2019-11-15 14:55:26 -05:00
Skye Wanderman-Milne
90093b7824 Remove version restriction from opt_einsum. 
See https://github.com/dgasmith/opt_einsum/issues/98.
 2019-08-23 14:43:52 -07:00
Skye Wanderman-Milne
921096e32e Require opt_einsum version to be less than 3.0.0. 
opt_einsum 3.0.0 adds a jax backend, which raises an exception on import.
 2019-08-19 19:28:07 -07:00
Peter Hawkins
8e66d29c45 Suppress flake8 warning from __version__ logic. 2019-08-04 12:12:53 -04:00
Peter Hawkins
08013954a4 Use fastcache for LRU caches in JAX. 
fastcache is both a faster cache implementation and is also thread-safe.
 2019-07-22 17:24:10 -04:00
Matthew Johnson
299977eeef exclude examples dir from setup.py find_packages 
fixes #582
 2019-04-06 14:16:57 -07:00
Peter Hawkins
f939ac078d Update XLA. 
Updates XLA to 00afc7bb81.

The new XLA release removes the use of protocol buffers from the XLA client. Fixes #349.
Add backward compatibility shims to jaxlib to allow older jax releases to still work on an up to date jaxlib.

The new XLA release also incorporates a fix that avoids a host-device copy for every iteration of a `lax.fori_loop()` on GPU. Fixes #402.

Add a new jaxlib.__version__ field, change jax/jaxlib compatibility logic to check for it.
 2019-02-26 06:07:44 -08:00
Matthew Johnson
9a9c304644 add version attribute 
following idea 3 here:
https://packaging.python.org/guides/single-sourcing-package-version/
 2019-02-13 20:04:38 -08:00
Matthew Johnson
9cd28d12a1 bump version for pypi 2019-02-13 14:55:23 -08:00
Matthew Johnson
a75d1c6e08 bump version for pypi 2019-02-05 19:06:29 -08:00
Matthew Johnson
76decd2ade version bump for pypi 2019-02-03 15:03:22 -08:00
Peter Hawkins
7d78c3c84c Drop jaxlib dependency from setup.py, add a requirements.txt for readthedoc build instead. 2019-01-16 09:45:25 -05:00
Peter Hawkins
86d8915c3d Add Sphinx-generated reference documentation for JAX. 2019-01-16 09:13:31 -05:00
Matthew Johnson
2166d3f269 bump version number for pypi 2019-01-12 08:06:20 -08:00
Matthew Johnson
95a86cd0fb bump version number for pypi 2018-12-24 11:34:38 -08:00
Matthew Johnson
9c6f4915e3 bump version number for pypi 2018-12-19 09:42:15 -08:00
Matthew Johnson
ae4bb8d346 bump version number for pypi 2018-12-19 09:08:37 -08:00
Matthew Johnson
308843f2d4 bump version number for pypi 2018-12-18 23:35:52 -08:00
Matthew Johnson
617dc18bc7 bump version for pypi 2018-12-15 20:16:07 -08:00
Matthew Johnson
fe059f3acc bump version number for pypi 2018-12-14 19:56:45 -08:00
Matthew Johnson
756bc8bb34 bump version number for pypi 2018-12-12 20:31:25 -08:00
Matthew Johnson
a7b339295e bump version number for pypi 2018-12-12 10:01:40 -08:00
Matthew Johnson
42302fc868 bump version number for pypi 2018-12-11 16:29:10 -08:00
Matthew Johnson
3ac1001c49 use find_packages() in setup.py 2018-12-11 15:45:56 -08:00
Matthew Johnson
4039aa480d bump version number for pypi 2018-12-11 13:32:22 -08:00
Colin
2bbc046d9d Require protobuf 3.6.0 or later 2018-12-09 10:11:21 -08:00
Matthew Johnson
4d6e7c8900 bump version number for pypi 2018-12-09 07:45:12 -08:00
Matthew Johnson
1469e82361 bump version number for pypi 2018-12-09 06:53:22 -08:00
Matthew Johnson
70604f4309 add scipy.stats to setup.py (should use find_packages()) 2018-12-07 15:05:20 -08:00
Matthew Johnson
ee1728c2af bump version for pypi 2018-12-07 11:15:50 -08:00
Matthew Johnson
bbc92ce6eb
Split out jax and jaxlib packages (#11) 
factor out 'jaxlib' as separate package
 2018-12-06 21:35:03 -05:00
Roy Frostig
a954389d06 source sync 
PiperOrigin-RevId: 222189611
 2018-11-21 20:22:37 -08:00
Roy Frostig
a3619ca89d source sync 
PiperOrigin-RevId: 222153576
 2018-11-21 20:22:30 -08:00
Matthew Johnson
50038c07c8 fix build file issues 2018-11-19 20:18:31 -08:00
Roy Frostig
da2d53ad33 source sync 2018-11-19 13:29:47 -08:00
Matthew Johnson
0dfa736ece make setup.py copy over .so files 2018-11-18 15:28:22 -08:00
Matthew Johnson
9584c4f55a add setup.py 2018-11-18 15:17:32 -08:00