🔥 A tool for visualizing and tracking your machine learning experiments. This repo contains the CLI and Python API.
-
Updated
Jul 16, 2024 - Python
🔥 A tool for visualizing and tracking your machine learning experiments. This repo contains the CLI and Python API.
Code I used for my YouTube videos
JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs welcome).
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Flax is a neural network library for JAX that is designed for flexibility.
Orbax provides common utility libraries for JAX users.
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
Convert ML Code Between Frameworks
Digital-analog quantum programming interface
PennyLane is a cross-platform Python library for quantum computing, quantum machine learning, and quantum chemistry. Train a quantum computer the same way as a neural network.
A retargetable MLIR-based machine learning compiler and runtime toolkit.
A JAX research toolkit for building, editing, and visualizing neural networks.
A JIT compiler for hybrid quantum programs in PennyLane
Implementation of PSGD optimizer in JAX
TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs, to provide better performance with lower memory utilization in both training and inference.
Add a description, image, and links to the jax topic page so that developers can more easily learn about it.
To associate your repository with the jax topic, visit your repo's landing page and select "manage topics."