Results for “jax”
8 skills
majiayu000 Jax
High-performance numerical computing with JAX, covering functional transformations, Flax NNX, and best practices for ML research.
567 · bundle
diegosouzapw Jax
Provides guidance on using JAX for machine learning and mathematical analysis, covering core concepts, transformations, ML specifics, control flow, and parallelism.
54 · bundle
schattenspiegel Numpyro Python
Write, debug, and test NumPyro probabilistic programs on JAX with correct shapes, PRNG keys, and inference choice.
0 · bundle
More results
qhjqhj00 Pennylane
Train quantum circuits with automatic differentiation and build hybrid quantum-classical models using PennyLane, including VQE, QAOA, and integration with PyTorch, JAX, and TensorFlow.
3 · bundle
orchestra-research Fine Tuning Serving Openpi
Fine-tune and serve Physical Intelligence OpenPI models (pi0, pi0-fast, pi0.5) using JAX or PyTorch backends for robot policy inference across ALOHA, DROID, and LIBERO environments.
10.4k · bundle
diegosouzapw Mhc
Implements Manifold-Constrained Hyper-Connections (mHC) using Doubly Stochastic Matrices to improve deep learning stability.
54 · bundle
k-dense-ai Pennylane
Train quantum circuits like neural networks with automatic differentiation, device-independent programming, and integration with PyTorch or JAX.
30.2k · bundle
antigravity Ml Engineer
Build production ML systems with PyTorch 2.x, TensorFlow, and modern ML frameworks, including model serving, feature engineering, A/B testing, and monitoring.
42.4k