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 3 comments of David Devoogdt

[call_tf] implementation of batching rule for vmap

I don't really need a round trip, just code to convert a tensorflow function to to jax function f: R^(nx3)->R^m (i.e a m dimensional function of n 3d molecular coordinates)....

[call_tf] implementation of batching rule for vmap

Seems good to me. Another option would be to add a configurable flag `SLOW_TF_VMAP = False`. If the flag is not set to `True`, raise an error within batching function...

[call_tf] implementation of batching rule for vmap

Some parts of my code were annoyingly slow, so I've reimplemented the batcher using the `tf.vectorized_map` function. Basically I redo the call_tf on the vectorized tensoflow function and pipe the...