jax-ml/jax
Python library for high-performance numerical computing and automatic differentiation on CPUs, GPUs, and TPUs
View on index · View in 3D Map
Related repos
- FastMCP framework for MCP servers in Python
- Python framework for stateful stream processing
- Folia—Paper fork for Minecraft server multithreading with region-based processing
- cloud native Java framework for fast microservices and containers
- Big Vision: codebase for training large-scale vision models using Jax/Flax andTPUs
- JAX library for efficient block Gibbs sampling
- JAX code and weights for running the 314B parameter Grok-1 language model
- open source platform for automating workflows and connecting APIs
- framework for building MCP servers with Python
- massively parallel programming language for GPUs and CPUs
