7 Commits

Author SHA1 Message Date
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
Jonas Rauber
8348856f92 added mypy to gitignore 2019-01-29 22:18:10 +01:00
vishwakftw
46f7c6a8a3 Add build/ and dist/ to .gitignore
An alternative to installing `jax` using `pip install -e .` is to use `python setup.py`. This creates build/ and dist/ directories that should not be included in diffs since they are meant to be local.

This PR adds those entries to the .gitignore file
2018-12-12 20:55:38 +05:30
Matthew Johnson
bbc92ce6eb
Split out jax and jaxlib packages (#11)
factor out 'jaxlib' as separate package
2018-12-06 21:35:03 -05:00
Alex Wiltschko
47ade41368 Adding quickstart notebook, and corresponding gitignore rules 2018-12-06 08:53:44 -05:00
Matthew Johnson
1b836515c2 update gitignore 2018-11-18 15:17:53 -08:00
Matthew Johnson
f78c05195d add gpu build script 2018-11-18 14:24:09 -08:00