-
Notifications
You must be signed in to change notification settings - Fork 27
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
TensorFlow backend #3
Comments
So finally TF also makes a step into the right direction by having essentially a (limited) numpy API available as well. The question is somewhat about numpy (and numba): do we expect to have inplace operations in the numpy implementation actually? If not TF, PyTorch and np can be implemented in one way, otherwise TF and PyTorch may need a separate, yet common to both, implementation. TF does not support the ufuncs though, this would result in a numpy array being returned. It just supports a |
Great! I think the current plan is to avoid in-place operations. Possibly there could be some room for it in the future, but not something we will target in version 1, and maybe never, since the best way to optimize may be to use numba/tf/pytorch. I'm doing some quick work for the Princeton flowmeter project, then I'll finish a fairly large commit including more functionality and updates. |
(I'm following the Uproot/awkward deadlines, so an initial release is targeting Dec 1) |
Great, that's good to hear!Looking forward to it; I'll gladly add the PyTorch/TF backends then |
Yep, looking forward to seeing what you have been preparing there :-). |
The Protocol in src/vector/protocols/lorentz.py should help with this, as well as the parametrized tests in tests/test_simple_param.py (static link, will likely rename soon). |
Work on TensorFlow backend.
The text was updated successfully, but these errors were encountered: