Skip to content
View mahdi-shafiei's full-sized avatar

Highlights

  • Pro

Block or report mahdi-shafiei

Block user

Prevent this user from interacting with your repositories and sending you notifications. Learn more about blocking users.

You must be logged in to block users.

Maximum 250 characters. Please don't include any personal information such as legal names or email addresses. Markdown supported. This note will be visible to only you.
Report abuse

Contact GitHub support about this user’s behavior. Learn more about reporting abuse.

Report abuse

Pinned Loading

  1. avitai/datarax avitai/datarax Public

    A Differentiable Data Pipeline Framework for JAX

    Python 1

  2. avitai/artifex avitai/artifex Public

    A research-focused modular generative modeling library built on JAX/Flax NNX

    Python 1

  3. avitai/opifex avitai/opifex Public

    A unified scientific machine learning framework built on JAX/Flax NNX

    Python 2

  4. avitai/calibrax avitai/calibrax Public

    Unified benchmarking and profiling framework for the JAX scientific ML ecosystem. Timing, GPU/energy monitoring, FLOPS counting, roofline analysis, statistical testing, regression detection, and CI…

    Python 1

  5. titans-flax titans-flax Public

    Titans: Learning to Memorize at Test Time

    Python 1