🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
-
Updated
Jul 3, 2024 - Python
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
Deep Learning for humans
NAACL '24 (Demo) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inference
Orbax provides common utility libraries for JAX users.
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
LinkOrgs: An R package for linking linking records on organizations using half a billion open-collaborated records from LinkedIn
Differentiable Fluid Dynamics Package
A differentiable visibility simulator based on spherical harmonics
A retargetable MLIR-based machine learning compiler and runtime toolkit.
A JAX Implementation of the Descript Audio Codec
Astronomical time series analysis with JAX
PEP 503 repository index for jax[cuda]
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.
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 tool for visualizing and tracking your machine learning experiments. This repo contains the CLI and Python API.
JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs welcome).
Flax is a neural network library for JAX that is designed for flexibility.
Galactic and Gravitational Dynamics in Python (+ GPU and autodiff)
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."