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).
-
Updated
May 20, 2024 - Python
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).
Machine learning algorithms for many-body quantum systems
A JAX Implementation of the Descript Audio Codec
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
⚡️SwanLab: your ML experiment notebook. 你的AI实验笔记本,跟踪与可视化你的机器学习全流程
GPU-accelerated NeuroEvolution of Augmenting Topologies (NEAT)
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
A differentiable physics engine and multibody dynamics library for control and robot learning.
Flax is a neural network library for JAX that is designed for flexibility.
A retargetable MLIR-based machine learning compiler and runtime toolkit.
Deep Learning for humans
A high-performance ML model serving framework, offers dynamic batching and CPU/GPU pipelines to fully exploit your compute machine
🔥 A tool for visualizing and tracking your machine learning experiments. This repo contains the CLI and Python API.
The Unified AI Framework
PEP 503 repository index for jax[cuda]
Notebooks 📔
dataset, train, 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."