Andrey Portnoy
74da736e0e
Update link to algebraic_simplifier.cc to point to OpenXLA instead of TF
2024-11-04 18:02:44 -05:00
Michael Hudgins
d4d1518c3d
Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax
...
PiperOrigin-RevId: 676843138
2024-09-20 07:52:33 -07:00
Roy Frostig
3c223cd253
docs: tidy up titles and headings
...
This shortens some titles and makes them more consistent. It also
removes "JAX" from several titles ("in JAX", "for JAX", "JAX's",
etc.). Since these are JAX docs, that ought to be clear from context.
2024-08-13 11:53:57 -07:00
Roy Frostig
2644299f7e
docs: sentence case index and sub-index headings
...
We currently use both forms, so for consistency (and easier reading),
pick this one.
2024-08-12 13:52:43 -07:00
Marco Selvi
7a8bcf6ee5
Fix parenthesis in "Gradients contain NaN where using where"
2024-05-31 12:03:39 +01:00
jax authors
7fb27a4cce
Merge pull request #19604 from Micky774:faq_cuda
...
PiperOrigin-RevId: 624234866
2024-04-12 11:09:12 -07:00
rajasekharporeddy
d92e735001
Fix Typos in docs
2024-04-11 07:54:13 +05:30
Meekail Zain
6b97a319e6
Updated FAQ to include section on CUDA library initialization failures
2024-04-04 23:08:10 +00:00
parmentelat
ed9a864a62
micro typo in faq.rst
2024-03-18 12:05:53 +01:00
rajasekharporeddy
75da54e9f4
Fix doc typos
2024-03-13 14:48:31 +05:30
helpingstar
4297e0a24c
fix typo2
2024-02-16 20:14:26 +09:00
helpingstar
fbccdb29f8
fix typo
2024-02-16 17:30:04 +09:00
yixiaoer
4e5b28a104
"fix typo"
2024-01-08 19:54:06 +08:00
Jake VanderPlas
97beb01c43
Deprecate the device() method of JAX arrays
2023-11-30 11:43:02 -08:00
Jake VanderPlas
0aec40a16f
Deprecate arr.device_buffer and arr.device_buffers
2023-11-29 15:31:01 -08:00
carlosgmartin
9f8e1bc34a
Add nn.squareplus.
2023-11-14 23:52:41 -05:00
Jake VanderPlas
4a5bd9e046
Fix typos across the package
2023-09-22 14:54:31 -07:00
8bitmp3
6cabc364b3
Update JAX F.A.Q.
2023-08-15 20:09:09 +00:00
Jake VanderPlas
d7a19442b6
DOC: fix formatting in FAQ
2023-06-14 03:17:08 -07:00
Jake VanderPlas
a083ba7853
DOC: explicitly mention io_callback in FAQ
2023-04-19 12:30:53 -07:00
Matthew Johnson
3eb9c7a6e7
update docs to remove stale reference to laziness optimization
2023-03-13 10:17:46 -07:00
Ikko Eltociear Ashimine
9be04e75fa
DOC: fix typo in faq.rst
...
maximium -> maximum
2023-01-31 17:47:07 +09:00
Jake VanderPlas
64621cc43f
FAQ: fix discussion of marking self as static
2023-01-17 12:03:49 -08:00
Jake VanderPlas
1bb4de280e
DOC: add FAQ entry on converting a tracer to an array
2023-01-10 12:28:16 -08:00
jax authors
4f7bf51ae6
Merge pull request #13328 from eltociear:patch-2
...
PiperOrigin-RevId: 494912964
2022-12-12 21:22:33 -08:00
Peter Hawkins
1bbcec79c1
Update FAQ since buffer donation is implemented on CPU.
...
PiperOrigin-RevId: 493372426
2022-12-06 11:57:34 -08:00
Ikko Ashimine
2b06888bdc
Fix typo in faq.rst
...
occured -> occurred
2022-11-21 02:44:36 +09:00
Jake VanderPlas
e9194b26b0
FAQ: fix JIT numerics discussion
2022-10-26 11:30:17 -07:00
jbushago
2038988783
Fix typo in faq.rst.
...
Fixed a small typo in the FAQ: "inthe" -> "in the".
2022-09-30 14:14:05 -04:00
Misha Brukman
86a1a6b1ee
Fix sample code block formatting
...
Add missing colon to format sample code as a code block rather than a quote.
2022-08-08 12:41:01 -04:00
George Necula
2fd46d13cd
Delete the masking.py
2022-07-25 11:25:29 +03:00
atgctg
1d5fbeadd2
Fix typo
2022-07-18 11:23:21 +02:00
Jake VanderPlas
fbb4eb7ace
DOC: add FAQ section on zero gradients for rank-based operations
2022-06-09 15:05:04 -07:00
jax authors
ea54754c49
Merge pull request #9118 from skye:device_context_manager
...
PiperOrigin-RevId: 452570041
2022-06-02 10:33:53 -07:00
Lena Martens
f8f5a5dca3
Add notes in buffer donation FAQ about key-word args limitation.
2022-05-25 15:33:04 +01:00
Onno Eberhard
7276731e8c
Small fix in FAQ documentation
2022-05-10 13:04:49 +02:00
Skye Wanderman-Milne
f26b866e08
Add jax.default_device context manager
...
This currently only supports setting a specific Device object, not a
platform like "cpu". That should be added in the future.
Bumps the minimum jaxlib version in order to include
https://github.com/tensorflow/tensorflow/pull/53656
2022-05-07 00:31:00 +00:00
Jake VanderPlas
c4836aa507
DOC: add FAQ entry on jit-compiling methods
2022-05-06 12:39:12 -07:00
George Necula
83b818d45c
Add more documentation for buffer donation
...
Fixes : #9237
2022-01-24 09:33:08 +01:00
Jake VanderPlas
b889282f6d
docs: add FAQ section about jit compilation & numerics
2021-12-28 08:57:51 -08:00
Jake VanderPlas
1f7d6316c2
doc: move stub section to bottom of FAQ
2021-12-15 16:19:14 -08:00
Jake VanderPlas
4a72e57ce0
DOC: add FAQ section on JAX vs. Numpy performance
2021-11-24 12:04:02 -08:00
elliotwaite
7392a57b75
DOC: many small fixes
2021-08-04 16:55:13 -07:00
George Necula
6a48c60a72
Rename master to main in embedded links.
...
Tried to avoid the change on external links to repos that
have not yet renamed master.
2021-06-18 10:00:01 +03:00
Jake VanderPlas
b88bf37698
DOC: fix typos in FAQ (np -> jnp)
2021-06-15 09:38:05 -07:00
Jake VanderPlas
12c84e7a50
Add jax.errors submodule & error troubleshooting docs
2021-03-03 12:39:12 -08:00
Stephan Hoyer
ac9b1ce921
Fix data placement reference
2021-03-01 10:12:19 -08:00
Stephan Hoyer
46db7bbe3b
DOC: Add a FAQ on "Benchmarking JAX code"
...
This should a useful reference for users surprised by how JAX is slow :)
2021-02-28 15:26:53 -08:00
8bitmp3
162a6f155a
Nit: lint (grammar) JAX F.A.Q.
2020-12-29 18:09:13 +00:00
Sharad Vikram
a51afa309c
Fix typo in FAQ
2020-11-09 21:06:53 -08:00
