TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor contents easier.
-
Updated
Jun 18, 2024 - Python
TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor contents easier.
Machine learning algorithms for many-body quantum systems
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural network models (and their initializations) to make them easier to train.
🔥 A tool for visualizing and tracking your machine learning experiments. This repo contains the CLI and Python API.
skscope: Sparse-Constrained OPtimization via itErative-solvers
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Digital-analog quantum programming interface
⚡️SwanLab: your ML experiment notebook. 你的AI实验笔记本,跟踪与可视化你的机器学习全流程
A retargetable MLIR-based machine learning compiler and runtime toolkit.
Deep Learning for humans
A JIT compiler for hybrid quantum programs in PennyLane
Differentiable and GPU accelerated scattering covariance statistics on the sphere
TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...
Flax is a neural network library for JAX that is designed for flexibility.
GPU-accelerated NeuroEvolution of Augmenting Topologies (NEAT)
Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and custom information operators. Compatible with the broader JAX scientific computing ecosystem.
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."