Jake Vanderplas

Line style

1

We should use the MarkConfig [stroke style](https://vega.github.io/vega-lite/docs/mark.html#stroke-style) to match pandas linestyle.

Add more local datasets

2

We can add local datasets if - the dataset license is compatible with the package MIT license (this is often tough to figure out, because the provenance of many available...

Add initial implementation of jax.Array base class

4

Part of #12049 The idea here is that `jax.Array` will eventually be defined in C++ and used as a true base class of both `core.Tracer` and `jax.experimental.array.Array`, which will be...

Fixes #10813

Strange behavior of `convert_element_type` on main branch

2

Running from github HEAD: ```python import jax.numpy as jnp import jax jax.config.update('jax_enable_x64', True) x = jnp.float64(2.718) x_f16 = x.astype('float16') print(x, x_f16) # 2.718 -15790.0 ``` version info: ```python jax.print_environment_info() #...

jax.random: warn on unsupported dtypes

1

Fixes #12364 Explanation for the default changes: `jnp.float_` and `jnp.int_` are currently equal to `float64` and `int64`, respectively, and are cannonicalized upon use. We can't use them as default arguments...

`pandax`! This is an experimental pandas-like API built on top of JAX. Very experimental work in progress. Related to #3009

jet of dynamic slice does not match jet of gather

1

This is blocking #12219 For example: ```python import jax.numpy as jnp from jax import lax from jax.experimental.jet import jet f_gather = lambda x: x[1:] f_dynamic_slice = lambda x: lax.dynamic_slice(x, (1,),...

generate dynamic_slice rather than slice for simple indexing/slicing

6

Followup to #11867, fixes #12198

jax.Array: support duck-typed isinstance checks

4

Why? As part of the type promotion discussion in JAX (#11859 & #12049), @patrick-kidger and I have been discussing what API jax should support for both annotations and `isinstance` checks;...