Kaylee de Soto

icon indicating an email [email protected]

icon company Harvard University icon location Cambridge, MA icon location Graduate student in Astronomy & Astrophysics

Results 3 issues of Kaylee de Soto

jax.lax.scan() segmentation fault with jax-metal on Mac M1

12 comment

### Description Segmentation fault when calling jax.lax.scan(), jax.lax.map(), and related functions. Segmentation fault can be traced back to core.AxisPrimitive().bind() call. Reproducible using jax-metal=0.0.5 and jax-metal=0.0.4, and using either M1 or...

bug
Apple GPU (Metal) plugin

BUG: .initialize_new_project.sh fails with strict mypy when including benchmark files

example_benchmark files need to be strictly typed by default for this to work. (same with any example files if enabled)

Add GNU GPLv3 License as option

License here: https://choosealicense.com/licenses/gpl-3.0/