jax-ml/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Explore 5 GitHub repositories focused on jax. Discover top-starred projects and those trending this week.
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
A theoretical reconstruction of the Claude Mythos architecture, built from first principles using the available research literature.
JAX-based neural network library
Distributed AI Model Training and LLM Fine-Tuning on Kubernetes
From teacher to tiles — a from-scratch LLM distillation & serving engine: custom Triton/CUDA kernels, FSDP distillation, paged-KV continuous batching, speculative decoding, a Rust gateway, a JAX oracle, and interpretability tooling.
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
A theoretical reconstruction of the Claude Mythos architecture, built from first principles using the available research literature.
JAX-based neural network library
Distributed AI Model Training and LLM Fine-Tuning on Kubernetes
From teacher to tiles — a from-scratch LLM distillation & serving engine: custom Triton/CUDA kernels, FSDP distillation, paged-KV continuous batching, speculative decoding, a Rust gateway, a JAX oracle, and interpretability tooling.