NGC-Learn: Neurobiological Learning and Biomimetic Systems Simulation in Python
-
Updated
May 31, 2024 - Python
NGC-Learn: Neurobiological Learning and Biomimetic Systems Simulation in Python
Distributed ML Training and Fine-Tuning on Kubernetes
Orbax provides common utility libraries for JAX users.
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
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
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.
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
A retargetable MLIR-based machine learning compiler and runtime toolkit.
🔥 A tool for visualizing and tracking your machine learning experiments. This repo contains the CLI and Python API.
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
Flax is a neural network library for JAX that is designed for flexibility.
TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...
Collection of eclectic utils for python.
This repository contains supporting code for the paper "Selecting a conceptual hydrological model using Bayes' factors computed with Replica Exchange Hamiltonian Monte Carlo" by Mingo et al.
⚡️SwanLab: your ML experiment notebook. 你的AI实验笔记本,跟踪与可视化你的机器学习全流程
Accelerate your training with this open-source library. Optimize performance with streamlined training and serving options with JAX. 🚀
Fully JAX-compatible python implementation of the DC-EGM algorithm from Iskhakov, Jorgensen, Rust, and Schjerning (QE, 2017).
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."