Question

Efficiently fill an array from a function

score 1.3k · Answer 1 · 2020-10-06 08:07:43

1.3k

Views

Efficiently fill an array from a function

Published on 06 October 2020 at 08:07

score 4.3k · Answer 2 · 2020-10-18 05:47:44

4.3k

Views

Conditional update in JAX?

Published on 18 October 2020 at 05:47

score 189 · Answer 3 · 2020-10-20 01:48:56

189

Views

Alternative to scipy stats zmap function

Published on 20 October 2020 at 01:48

score 74 · Answer 4 · 2023-10-16 14:09:06

74

Views

Installing Jax on Fedora 38

Published on 16 October 2023 at 14:09

score 40 · Answer 5 · 2023-10-10 19:50:40

40

Views

Are PINNs inherently worse at solving 2nd-order PDEs?

Published on 10 October 2023 at 19:50

score 83 · Answer 6 · 2023-10-20 04:15:52

83

Views

how to make hashable datatype to use with jit?

Published on 20 October 2023 at 04:15

score 105 · Answer 7 · 2023-10-22 03:05:23

105

Views

better way to use lax.cond, partial, vmap

Published on 22 October 2023 at 03:05

score 169 · Answer 8 · 2023-10-21 01:01:12

169

Views

Why does jnp.einsum produce a different result from manual looping?

Published on 21 October 2023 at 01:01

score 176 · Answer 9 · 2023-10-25 13:26:29

176

Views

vectorized minimization and root finding in jax

Published on 25 October 2023 at 13:26

score 142 · Answer 10 · 2023-10-24 18:31:23

142

Views

Appending to list alternative workflow for JAX

Published on 24 October 2023 at 18:31

score 70 · Answer 11 · 2023-10-20 21:56:48

70

Views

how to use jit with control flow and arrays without slowing down

Published on 20 October 2023 at 21:56

score 73 · Answer 12 · 2023-10-23 10:01:13

73

Views

Cross dimensional segmented operation

Published on 23 October 2023 at 10:01

score 205 · Answer 13 · 2023-10-25 20:15:49

205

Views

Why matrix multiplication results with JAX are different if the data is sharded differently on the GPU

Published on 25 October 2023 at 20:15

score 165 · Answer 14 · 2023-10-28 21:50:08

165

Views

Idiomatic ways to handle errors in JAX jitted functions

Published on 28 October 2023 at 21:50

score 2k · Answer 15 · 2023-10-29 00:25:24

2k

Views

CUDA and JAX libraries are not compatible

Published on 29 October 2023 at 00:25

score 239 · Answer 16 · 2023-10-25 16:35:09

239

Views

PyTorch autograd: Efficient computation of Jacobian and Jacobian-Vector-product of scalar function over range of inputs

Published on 25 October 2023 at 16:35

score 159 · Answer 17 · 2023-10-26 03:47:51

159

Views

Convert for loop to jax.lax.scan

Published on 26 October 2023 at 03:47

score 194 · Answer 18 · 2023-11-05 22:33:11

194

Views

JAX vmap JIT behind the scenes?

Published on 05 November 2023 at 22:33

score 257 · Answer 19 · 2023-11-07 14:51:43

257

Views

JAX @jit for nested class method

Published on 07 November 2023 at 14:51

score 84 · Answer 20 · 2023-11-09 10:44:03

84

Views

Mapping over multiple functions in JAX to speed up for loops

Published on 09 November 2023 at 10:44

List Question

Efficiently fill an array from a function

Conditional update in JAX?

Alternative to scipy stats zmap function

Installing Jax on Fedora 38

Are PINNs inherently worse at solving 2nd-order PDEs?

how to make hashable datatype to use with jit?

better way to use lax.cond, partial, vmap

Why does jnp.einsum produce a different result from manual looping?

vectorized minimization and root finding in jax

Appending to list alternative workflow for JAX

how to use jit with control flow and arrays without slowing down

Cross dimensional segmented operation

Why matrix multiplication results with JAX are different if the data is sharded differently on the GPU

Idiomatic ways to handle errors in JAX jitted functions

CUDA and JAX libraries are not compatible

PyTorch autograd: Efficient computation of Jacobian and Jacobian-Vector-product of scalar function over range of inputs

Convert for loop to jax.lax.scan

JAX vmap JIT behind the scenes?

JAX @jit for nested class method

Mapping over multiple functions in JAX to speed up for loops

Trending Questions

Popular # Hahtags

Popular Questions