GitHub - jax-ml/jax: Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
8d ago· 6 min readCode
Source
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more - jax-ml/jax
You might also wanna read
jax-js: A JavaScript Implementation of Google's JAX Machine Learning Framework for Web Browsers
The article announces jax-js, a new machine learning library for the web that reimplements Google DeepMind's JAX framework in pure JavaScrip
mytorch: Python Automatic Differentiation Library Inspired by PyTorch
mytorch is an open-source Python library that implements automatic differentiation with a PyTorch-like API, using NumPy for computations. Th
Microjax: A Simplified Autograd Engine with Jax-like API
Microjax is a tiny autograd engine with a Jax-like API inspired by Micrograd and JAX, offering a functional style for machine learning. The
Jasmine: A Scalable JAX-Based World Modeling Codebase for Efficient AI Training
Researchers introduce Jasmine, a high-performance JAX-based world modeling codebase designed to scale from single hosts to hundreds of accel
Porting JustHTML HTML5 Parser from Python to JavaScript Using Codex CLI and GPT-5.2
The author describes successfully porting JustHTML, a standards-compliant HTML5 parser originally written in Python, to JavaScript using Cod
simonwillison.net·6mo ago
GPEmu: A GPU Emulator for Rapid, Low-Cost Deep Learning Prototyping [pdf]
vldb.org·1y ago

Comments
Sign in to join the conversation.
No comments yet. Be the first.