mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 04:16:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
1,439 Commits 503 Branches 263 Tags
Commit Graph

63 Commits

Author SHA1 Message Date
Peter Hawkins
f33629c13b Update README.md to add a quick link banner, and to recommend pytest-xdist. 2019-04-16 09:47:26 -04:00
Matthew Johnson
6b5af62b81
fix broken link in readme 2019-04-15 07:54:46 -07:00
Matthew Johnson
3a633dbdb4 mention lack of windows support in readme (#612) 2019-04-15 07:07:58 -07:00
Matthew Johnson
0cf14837c9 make a lax package, revert control flow names (#607) 
c.f. #597
pair=skyewm
 2019-04-12 16:28:40 -07:00
Skye Wanderman-Milne
105e46f379 Factor out control flow from lax.py into lax_control_flow.py. 
Also moves control flow tests to lax_control_flow_test.py.
 2019-04-12 13:57:18 -07:00
Matthew Johnson
027c94da4c
update readme gotchas section 2019-04-05 12:59:52 -07:00
Jamie Townsend
1c9b9a57fd Use jax.random for stax initialization 2019-04-03 12:54:02 +01:00
Anselm Levskaya
dad8157d7c gotchas updates and README.md colab link fixes 2019-03-27 13:24:05 -07:00
Peter Hawkins
4fd21867c1 Update Jaxlib versions in documentation/notebooks. 2019-03-27 15:45:03 -04:00
Matthew Johnson
cefbea6a42
README should point to hosted colab for gotchas 2019-03-24 08:50:56 -07:00
Matthew Johnson
a848d0c0fe
update readme, link gotchas notebook 2019-03-24 07:59:50 -07:00
Alex Wiltschko
ab486bd551
Update README.md 2019-03-13 13:42:38 -04:00
Bharat Raghunathan
bd7d874ac7
Update README.md to avoid overflow 2019-03-03 00:17:17 +05:30
Bharat Raghunathan
0c7aca1b31
Fix #396 by updating the comment in README.md 2019-03-02 23:49:35 +05:30
Peter Hawkins
500ca34fde Update jaxlib version numbers in docs/notebooks. 2019-03-01 10:05:50 -05:00
Peter Hawkins
901a5e5291 Update Jaxlib version to 0.1.8. 2019-02-17 15:22:11 -05:00
Matthew Johnson
70cc852491 readme: don't use np.float64(x) (closes #367) 2019-02-16 16:11:49 -08:00
Matthew Johnson
42b3218e90
Merge pull request #346 from gdahia/fix-readme-toy-example 
Fix README toy example incomplete `label_probs` var renaming
 2019-02-08 14:20:17 -08:00
Gabriel Dahia
197755b066 Fix README toy example incomplete var renaming 
Rename of `label_probs` into `label_logprobs` in the toy example was
incomplete and, as a consequence, the toy example no longer worked. This
renames `label_probs` completely and fixes the toy example.
 2019-02-08 16:26:52 -03:00
Matthew Johnson
0ca0968533
Merge pull request #338 from gdahia/fix-readme-toy-example 
Change implementation of negative log-likelihood in README toy example
 2019-02-08 11:13:55 -08:00
Gabriel Dahia
80bbdbbb75 Rename label_probs -> label_logprobs in README 2019-02-08 15:28:19 -03:00
Peter Hawkins
20c9737359 Update jaxlib references to 0.1.7. 
Enable einsum tests (issue #37).
 2019-02-08 11:48:32 -05:00
Gabriel Dahia
d319f5805b Fix toy example negative log-likelihood loss 2019-02-07 11:09:10 -03:00
Matthew Johnson
5847d96168 rename "minmax" -> "optimizers" 2019-02-06 11:02:16 -08:00
Peter Hawkins
1d9be64e01 Update Jaxlib references to 0.1.6. 2019-01-18 23:14:56 -05:00
Matthew Johnson
42bd253cbb add prng design doc in markdown format 2019-01-18 07:38:43 -08:00
Peter Hawkins
e062127d1e Update documentation and notebooks for Jaxlib 0.1.5. 2019-01-17 13:24:58 -05:00
Peter Hawkins
bfa35d26db Move link to reference documentation earlier. 
Add instructions to install scipy when building from source.
 2019-01-16 13:32:16 -05:00
Peter Hawkins
63b49e4cbc Add links to reference documentation from README.md. 2019-01-16 10:21:08 -05:00
Matthew Johnson
d9a798f9db
fix minor typos in readme.md 2019-01-02 12:51:26 -08:00
Matthew Johnson
b781e4509a update jaxlib references to 0.1.3 2018-12-19 08:42:29 -08:00
Matthew Johnson
6865783cf9 fix some longstanding readme typos 2018-12-18 16:31:51 -08:00
Matthew Johnson
6de5c8a698 add test-running instructions (fixes #67) 2018-12-14 16:48:08 -08:00
Matthew Johnson
aa7c8c367b tweak minmax example imports in readme 2018-12-14 10:35:01 -08:00
Matthew Johnson
87ee4b7c56 update jaxlib references to 0.1.2 2018-12-13 19:59:08 -08:00
Matthew Johnson
2048874d43 add '--upgrade' to pip install examples 2018-12-13 17:44:08 -08:00
Matthew Johnson
2e40e63527 update jaxlib wheel urls to 0.1.1 2018-12-13 17:35:19 -08:00
Matthew Johnson
e4ef7eae27 add travis badge to readme 2018-12-13 17:12:06 -08:00
Matthew Johnson
51d89a332c
tweak readme text 2018-12-12 21:32:30 -08:00
Matthew Johnson
3307553dbe
readme: JAX is about composable transformations 2018-12-12 19:01:32 -08:00
Matthew Johnson
650db099ef
tweak readme to make colab bullet fit on one line 2018-12-11 20:46:02 -08:00
Matthew Johnson
9099272e72 fix typo in readme 2018-12-11 12:55:39 -08:00
Matthew Johnson
cbd9ace425 change vmap api to be curried (closes #78) 2018-12-11 12:52:09 -08:00
Alex Wiltschko
e425f244db
Update README.md 2018-12-10 19:22:10 -05:00
Matthew Johnson
0f8710a6e3
add another link to Autograd in the readme 2018-12-10 07:13:51 -08:00
Matthew Johnson
5a604d5b14
add more contributors to readme 2018-12-09 01:24:38 -05:00
Dougal Maclaurin
9f61472a91
Update README.md 2018-12-08 11:20:40 -05:00
Dougal Maclaurin
0cd84efac1
Small changes to readme 2018-12-08 11:19:03 -05:00
Matthew Johnson
9d484b6d4c
attempt to center-justify the jax logo in readme 2018-12-08 08:05:16 -05:00
Doug Blank
13163ffefe
Typo, Python parens 2018-12-08 03:40:33 -05:00