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

perf: benchmarking our models against Jax (Flax) #1000

Open
wants to merge 8 commits into
base: main
Choose a base branch
from
Open

perf: benchmarking our models against Jax (Flax) #1000

wants to merge 8 commits into from
+371 −4

Commits on Nov 5, 2024

  1. fix: don't trace through the model

    @avik-pal
    avik-pal committed Nov 5, 2024
    Configuration menu
    Copy the full SHA
    501b5b6 View commit details
    Browse the repository at this point in the history

  2. perf: add comparisons to JAX

    @avik-pal
    avik-pal committed Nov 5, 2024
    Configuration menu
    Copy the full SHA
    6cbc020 View commit details
    Browse the repository at this point in the history

  3. perf: add direct Lux + CUDA numbers

    @avik-pal
    avik-pal committed Nov 5, 2024
    Configuration menu
    Copy the full SHA
    08c98c2 View commit details
    Browse the repository at this point in the history

  4. perf: force sync

    @avik-pal
    avik-pal committed Nov 5, 2024
    Configuration menu
    Copy the full SHA
    a2510ea View commit details
    Browse the repository at this point in the history

  5. perf: more performance numbers

    @avik-pal
    avik-pal committed Nov 5, 2024
    Configuration menu
    Copy the full SHA
    0537f57 View commit details
    Browse the repository at this point in the history

  6. perf: try AoT compile for jax

    @avik-pal
    avik-pal committed Nov 5, 2024
    Configuration menu
    Copy the full SHA
    8d2209e View commit details
    Browse the repository at this point in the history

  7. perf: add optimize option to reactant runs

    @avik-pal
    avik-pal committed Nov 5, 2024
    Configuration menu
    Copy the full SHA
    12dfdba View commit details
    Browse the repository at this point in the history

  8. fix: remove tracing overload 

    [skip ci] [skip docs] [skip benchmarks] [skip tests]
    @avik-pal
    avik-pal committed Nov 5, 2024
    Configuration menu
    Copy the full SHA
    b88f628 View commit details
    Browse the repository at this point in the history