jax.random.randint silently returns zero for large values of maxval better_errors good first issue

#2006 opened Jan 15, 2020 by vmikulik

10

Implement NumPy sorting routines enhancement good first issue

#2079 opened Jan 26, 2020 by shoyer

jnp.array(x, dtype=object) Causes Error(Shape Checking Issue) better_errors documentation good first issue

#2347 opened Mar 4, 2020 by Yeachan-Heo

2

Implement all numpy.fft operations enhancement good first issue

#1877 opened Dec 17, 2019 by shoyer

9

Support scipy.special.erfcx enhancement good first issue

#1987 opened Jan 12, 2020 by cossio

5

document jax_debug_nans better documentation

#2649 opened Apr 9, 2020 by mattjj

`jax.api.mask` Documentation and #292 documentation

#2643 opened Apr 8, 2020 by tbsexton

Performance cookbook documentation

#2940 opened May 3, 2020 by shoyer

3

RuntimeError: Resource temporarily unavailable due to running out of threads (ulimit -u) documentation question

#2685 opened Apr 11, 2020 by mgbukov

10

Documentation: `custom_jvp` functions not allowed to close over values documentation

#2676 opened Apr 10, 2020 by hexahedria

In-place updates and JAX immutable arrays documentation

#2219 opened Feb 12, 2020 by bringingjoy

1

Documentation of QR bugged documentation

#2254 opened Feb 18, 2020 by Jakob-Unfried

5

Add depthwise convolutions to stax enhancement good first issue

#1114 opened Aug 5, 2019 by untom

4

disable_jit should error if jit'd documentation

#2148 opened Feb 3, 2020 by johnmellor

3

Some usability problems for new users better_errors documentation

#2057 opened Jan 23, 2020 by skye

2

jax.lax.scan benefits apart from compilation time documentation

#1897 opened Dec 19, 2019 by clemisch

2

sp.ndimage.zoom / tf.image.resize_images alternative enhancement good first issue

#862 opened Jun 17, 2019 by romanngg

11

Document how to use keyword arguments with jit documentation

#1936 opened Jan 3, 2020 by shoyer

7

Add documentation for JAX foreign function interface documentation

#1554 opened Oct 22, 2019 by gnecula

4

Add a pmap cookbook documentation

#1487 opened Oct 10, 2019 by georgedahl

4

Fix sphinx warnings when generating the documentation documentation

#1433 opened Oct 3, 2019 by gnecula

Limit number of threads when running on CPU documentation

#1539 opened Oct 21, 2019 by dionhaefner

17

Add support for Jax on Windows build contributions welcome good first issue

#438 opened Feb 23, 2019 by dwang55

16

data type not understood on simple expression `np.exp([[[1.], [2.]]])` documentation

#1363 opened Sep 17, 2019 by brianwa84

2

Custom VJPs for external functions documentation

#1142 opened Aug 8, 2019 by mrbaozi

12

`is` keyword is not preserved under vmap. documentation

#1103 opened Aug 3, 2019 by sschoenholz

3

add examples and docs for how to use lax.conv functions enhancement good first issue

#224 opened Jan 11, 2019 by mattjj

2

random.normal is slow when shape is a list instead of a tuple documentation

#1038 opened Jul 19, 2019 by matthewdhoffman

4

Documentation for wrapping xla ops documentation

#1145 opened Aug 8, 2019 by gehring

4

Use of CUDA_VERSION in installation instructions in README.md

#2317 opened Feb 26, 2020 by adityaiitb

document memory management (DeviceArrays and where to find them) documentation enhancement

#863 opened Jun 17, 2019 by mattjj

1

lax.dynamic_slice inside jit documentation question

#1007 opened Jul 9, 2019 by shoyer

6

Unimplemented NumPy core functions enhancement good first issue

#70 opened Dec 11, 2018 by alexbw

26

Loading pre-trained ONNX models enhancement good first issue

#138 opened Dec 18, 2018 by alexbw

1

jax.numpy.subtract throws runtime error when one of the arrays is object type documentation

#856 opened Jun 14, 2019 by neha-wadia

3

Improved error messages and documentation regarding what can/cannot be `jit`ed documentation

#953 opened Jul 1, 2019 by samuela

6

lax.cond, vmap and data dependent while loops may cause non-termination bug

#2947 opened May 4, 2020 by lumip

4

Develop best practice for input pipelines documentation

#535 opened Mar 27, 2019 by skye

1

Add wrapped function to str/repr of custom_jvp enhancement

#2848 opened Apr 25, 2020 by tomhennigan

3

jax.numpy array indexing different from numpy documentation

#278 opened Jan 26, 2019 by jonasrauber

12

deepcopy of jax.numpy array generates regular numpy array

#2632 opened Apr 7, 2020 by durian888

3

document when and how to make ndarray copies of DeviceArrays documentation

#390 opened Feb 16, 2019 by mattjj

2

Inconsistent error messages when passing bad arguments to primitives better_errors

#2904 opened Apr 30, 2020 by gnecula

jvp docs names output v which is confuses with v in jvp

#1311 opened Sep 5, 2019 by jessebett

pytrees in jax.experimental.loops.Scope enhancement

#2767 opened Apr 20, 2020 by yingted

1

Sampling from random.normal in bfloat16 better_errors

#2274 opened Feb 20, 2020 by ajbrock

jax.random.randint raises an error if minval and maxval have different rank than len(shape) better_errors

#2836 opened Apr 25, 2020 by dwf

2

reverse direction fori_loop

#2016 opened Jan 16, 2020 by HHalva

3

[FR] add betaincinv enhancement

#2399 opened Mar 10, 2020 by tuchandra

4

XLA_in_Python notebook fails to build in RTD with XLA shape error

#2236 opened Feb 15, 2020 by gnecula

mypy and pytype for type hints

#2474 opened Mar 21, 2020 by gnecula

2

Implement matrix exponential enhancement

#1635 opened Nov 6, 2019 by jekbradbury

5

nan from the example notebooks if set param_scale to 0.1

#1365 opened Sep 17, 2019 by xysun

jitted functions can't return generic callables better_errors

#2152 opened Feb 3, 2020 by mattjj

1

isinstance(numpy.zeros(1), jax.numpy.ndarray) returns True bug question

#2014 opened Jan 16, 2020 by honnibal

9

jax.numpy functions `out` argument fails question

#1328 opened Sep 9, 2019 by mgbukov

2

Automatically treat dataclasses as pytrees

#2371 opened Mar 6, 2020 by shoyer

7

JAX argsort has different behavior on ties compared to {np,tf}.argsort question

#2124 opened Jan 29, 2020 by srvasude

5

Better error message for mismatched pmap sizes

#1382 opened Sep 21, 2019 by Thenerdstation

extreme int64 values in reduce-min cause failures when 64-bit mode is off bug

#984 opened Jul 6, 2019 by mattjj

List comprehensions and for-loops on xla.DeviceArrays return numpy.ndarrays question

#1583 opened Oct 28, 2019 by jessebett

5

Annealed Importance Sampling - numpy/numba/jax comparison application performance

#1421 opened Oct 2, 2019 by adamhaber

10

isinstance(1., np.ndarrray) = True if jitted

#1080 opened Aug 1, 2019 by romanngg

provide better hints in error message "abstract value passed to `bool`" enhancement

#905 opened Jun 22, 2019 by mattjj

print-debugging lax loops question

#999 opened Jul 8, 2019 by samuela

10

Entry function 'fusion_2' uses too much parameter space bug

#677 opened May 5, 2019 by guyvdbroeck

7