🔥 A tool for visualizing and tracking your machine learning experiments. This repo contains the CLI and Python API.
-
Updated
Jun 11, 2024 - Python
🔥 A tool for visualizing and tracking your machine learning experiments. This repo contains the CLI and Python API.
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.
Flax is a neural network library for JAX that is designed for flexibility.
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Deep Learning for humans
A model predictive controller for quadruped robots based on the single rigid body model and written in python. Gradient-based (acados) or Sampling-based (jax).
NGC-Learn: Neurobiological Learning and Biomimetic Systems Simulation in Python
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
A retargetable MLIR-based machine learning compiler and runtime toolkit.
Orbax provides common utility libraries for JAX users.
A JAX Implementation of the Descript Audio Codec
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
JAX compilation of RDDL description files, and a differentiable planner in JAX.
Galactic and Gravitational Dynamics in Python (+ GPU and autodiff)
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...
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."