Skip to content
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.

Sign up for GitHub

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

Jump to bottom

JIT-compile TensorFlow functions #440

Open
redeboer opened this issue May 24, 2022 · 0 comments · May be fixed by #441
Open

JIT-compile TensorFlow functions #440

redeboer opened this issue May 24, 2022 · 0 comments · May be fixed by #441
Labels
⚙️ Enhancement Improvements and optimizations of existing features

Comments

@redeboer
Copy link
Member

The TF backend currently does not decorate the lambdified function with tf.function nor jit_compile it.

tensorwaves/src/tensorwaves/function/sympy/__init__.py

Lines 237 to 243 in 876e40e

return _sympy_lambdify(
expression,
symbols,
modules=tnp,
printer=TensorflowPrinter(),
use_cse=use_cse,
)

@redeboer redeboer added the 🔨 Maintenance Maintenance and upkeep improvements label May 24, 2022
@redeboer redeboer self-assigned this May 24, 2022
@redeboer redeboer linked a pull request May 24, 2022 that will close this issue
Draft
@redeboer redeboer added ⚙️ Enhancement Improvements and optimizations of existing features and removed 🔨 Maintenance Maintenance and upkeep improvements labels Jan 6, 2023
@redeboer redeboer changed the title JIT-compile TF function JIT-compile TensorFlow functions Jan 9, 2024
@redeboer redeboer removed their assignment Jan 9, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
⚙️ Enhancement Improvements and optimizations of existing features
Projects
Student projects
Status: 📋 Backlog
Milestone
No milestone
Development

Successfully merging a pull request may close this issue.

ENH: jit-compile TF functions redeboer/ComPWA-tensorwaves
1 participant
@redeboer