10 Commits

Author	SHA1	Message	Date
George Necula	e42c010605	Create developer documentation. ... * Moved out of README.md some developer-only stuff to docs/developer.rst. * Added documentation about building the documentation	2019-10-09 17:24:01 +02:00
George Necula	75c2236063	Addressed comments for the Colab. ... * Cleaned up use of section levels * Renamed ma to multiply_add and sq_add to square_add * Other minor clarifications * Separated the Colabs into Tutorials and Advanced Tutorials	2019-10-03 11:20:04 +02:00
George Necula	454320e9c9	Added How_JAX_primitives_work colab	2019-10-02 14:42:01 +02:00
Stephan Hoyer	9bd7330e1f	Notebooks on RTD (#1121)	2019-09-30 11:00:02 -07:00
Matthew Johnson	3c2a73592c	improve rank promotion warning, add doc page	2019-08-25 14:28:53 -07:00
Peter Hawkins	6dc730a5f4	Make JAX tracer state thread-local. Allows performing traces in separate threads. ... Using threading within a traced context still won't work, but that is perhaps less important than the ability to call JIT-ted computations from separate threads. (Revives https://github.com/google/jax/pull/734.)	2019-08-09 13:55:20 -04:00
Peter Hawkins	bcacdfe315	Add some brief documentation about how to profile/trace JAX programs.	2019-08-08 21:02:41 -04:00
Skye Wanderman-Milne	5b9849f177	Add docs on GPU memory allocation.	2019-07-30 15:49:35 -07:00
Peter Hawkins	91e6ba322c	Add documentation on asynchronous dispatch.	2019-06-04 10:09:43 -04:00
Peter Hawkins	86d8915c3d	Add Sphinx-generated reference documentation for JAX.	2019-01-16 09:13:31 -05:00