← Explore
TOPIC

#jax

Open source repositories tagged with #jax, ranked by health score.

NVIDIA
NVIDIA/TransformerEngine
Python
88
health

A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit and 4-bit floating point (FP8 and FP4) precision on Hopper, Ada and Blackwell GPUs, to provide better performance with lower memory utilization in both training and inference.

★ 3.6k
wandb
wandb/wandb
Python
87
health

The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.

★ 11.3k
CogniPilot
CogniPilot/rumoca
Rust
85
health

a rust-native Modelica compiler for fast, composable, scientific computing

★ 135