mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 20:36:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
7,287 Commits 504 Branches 263 Tags
Commit Graph

50 Commits

Author SHA1 Message Date
Skye Wanderman-Milne
757247b791 Update README, etc. for jaxlib 0.1.63 release 2021-03-17 10:14:52 -07:00
Jake VanderPlas
d6408a4e6a Add extras_require to setup.py 2021-03-16 13:23:46 -07:00
George Necula
f105517ea2 Fixed mypy type errors for numpy 1.20 
Revert also previous changes that pinned numpy to 1.19.

One of the changes in numpy 1.20 is to add more type annotations.
However, this sometimes make mypy give errors. A common example is
numpy.take, which with the new type annotation does not appear to
mypy as indexable.

Another change is that np.int and np.bool are deprecated. One
should use np.bool_ or np.int_, or the built-ins bool and int.
 2021-02-05 10:40:47 +02:00
George Necula
a145e3d414 Pin numpy to max version 1.19, to avoid errors with 1.20 
Will fix the numpy errors separately.
 2021-01-31 15:18:54 +02:00
Peter Hawkins
328ddfca9f Add py.typed to setup.py for PEP 561 compliance. 2021-01-14 15:08:53 -05:00
Jake VanderPlas
5959fa9ccf Run main test suite under Python 3.8 2020-12-30 13:20:13 -08:00
John Aslanides
6029f02be8 Add setup classifiers fields (including supported Python versions) for PyPI. 
This is following https://packaging.python.org/tutorials/packaging-projects/.
 2020-12-30 11:29:34 +00:00
Srijan Saurav
40e20242db
Fix code quality issues (#4302) 
Changes:
- Fix unnecessary generator
- Iterate dictionary directly instead of calling .keys()
- Remove global statement at the module level
- Use list() instead of a list comprehension
- Use with statement to open the file
- Merge isinstance calls
 2020-09-17 09:21:18 -07:00
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