froystig
Follow
Pinned Loading
233 contributions in the last year
Less
No contributions.
Low contributions.
Medium-low contributions.
Medium-high contributions.
High contributions.
More
Contribution activity
April 2025
Created 15 commits in 1 repository
Created a pull request in jax-ml/jax that received 4 comments
add an out_sharding option to jax.random.randint
Drop into Auto mode in the implementation.
+35
−13
lines changed
•
4
comments
Opened 5 other pull requests in 1 repository
jax-ml/jax
5
merged
-
add an
out_shardingoption tojax.random.truncated_normalThis contribution was made on Apr 4 -
add an
out_shardingoption tojax.random.permutationThis contribution was made on Apr 3 -
add an
out_shardingoption tojax.random.bitsThis contribution was made on Apr 3 -
upgrade docs from
jax.coretojax.extend.corewhere needed to fix doc buildThis contribution was made on Apr 2 -
let XLA metadata be unset in nested dynamic scopes
This contribution was made on Apr 2
Reviewed 8 pull requests in 2 repositories
jax-ml/jax
7 pull requests
-
jax.random.bernoulli: use higher-resolution sampler
This contribution was made on Apr 15
-
Make
traceandlowerclass attributes forjax.jit.This contribution was made on Apr 9 -
Fix linear_call to allow recursive definitions
This contribution was made on Apr 9
-
Add JVP rule for linear_call
This contribution was made on Apr 8
-
[export] Add support for serializing functions with PRNG keys as inputs/outputs
This contribution was made on Apr 7
-
Fix jax.random.randint for 8-bit and 16-bit dtypes
This contribution was made on Apr 4
-
make random_gamma_grad not a primitive anymore
This contribution was made on Apr 1
jax-ml/jax-llm-examples
1 pull request
-
Index Pallas refs instead of using
pl.loadandpl.storeThis contribution was made on Apr 7
Opened 1 issue in 1 repository
jax-ml/jax
1
open
-
build RNG APIs on dtypes rather than "impls"
This contribution was made on Apr 9