Skip to content
@avitai

Avitai Bio

Popular repositories Loading

  1. opifex opifex Public

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

    Python 2

  2. datarax datarax Public

    A Differentiable Data Pipeline Framework for JAX

    Python 1

  3. artifex artifex Public

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

    Python 1

  4. calibrax 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

Repositories

Showing 4 of 4 repositories
  • datarax Public

    A Differentiable Data Pipeline Framework for JAX

    avitai/datarax’s past year of commit activity
    Python 1 MIT 0 0 0 Updated Mar 11, 2026
  • 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 integration.

    avitai/calibrax’s past year of commit activity
    Python 1 MIT 0 0 0 Updated Mar 10, 2026
  • artifex Public

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

    avitai/artifex’s past year of commit activity
    Python 1 MIT 0 0 0 Updated Mar 7, 2026
  • opifex Public

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

    avitai/opifex’s past year of commit activity
    Python 2 MIT 0 0 5 Updated Mar 3, 2026

People

This organization has no public members. You must be a member to see who’s a part of this organization.

Top languages

Python