David Devoogdt

icon location Belgium icon location Hello, I'm David Devoogdt student engineering physics at Ghent University. Besides physics and math I like computer science. I also love to sculpt in Blender3D.

Results 1 issues of David Devoogdt

[call_tf] implementation of batching rule for vmap

6 comment

Hi, I'm trying to use call_tf in combination with jacrev and hence vmap. The primitive_batcher of type call_tf_p is not implemented. I tried to modify the source code of `jax/experimental/jax2tf/call_tf.py`...

enhancement