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

People

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

Top languages

Loading…

Most used topics

Loading…