Explore Projects
Discover 54 open source projects
Showing 1-20 of 54 projects
keras-team/keras
Keras 3 is a multi-backend deep learning framework for building and training models with support for JAX, TensorFlow, PyTorch, and OpenVINO.
xai-org/grok-1
Open-source Grok-1 model for local inference with JAX
jax-ml/jax
JAX is a high-performance numerical computing library for Python, enabling transformations like differentiation and JIT compilation for GPUs/TPUs.
d2l-ai/d2l-en
Interactive deep learning book with code and math
lukasmasuch/best-of-ml-python
Curated list of top machine learning Python libraries, updated weekly with quality scores.
ivy-llc/ivy
Ivy is a Python library that simplifies converting machine learning code between frameworks like PyTorch, TensorFlow, and JAX.
wandb/wandb
The AI developer platform to train and fine-tune models, and manage models from experimentation to production.
OpenFeign/feign
Feign is a Java HTTP client library that makes writing HTTP clients easier by providing a declarative interface.
arogozhnikov/einops
A flexible and powerful tensor operations library for readable and reliable deep learning code in Python.
dropwizard/dropwizard
Dropwizard is a Java web framework for building production-ready RESTful services.
google/trax
Trax is a deep learning library that provides clear code and fast performance for building AI models.
google/flax
Flax is a flexible neural network library for JAX, designed for easy experimentation and research in machine learning.
probml/pyprobml
Python code for a book on probabilistic machine learning, with support for various ML frameworks.
kingoflolz/mesh-transformer-jax
Transformers built with JAX and Haiku for mesh-based AI applications.
BoltzmannEntropy/interviews.ai
This repository provides interview preparation resources for AI and machine learning developers.
sanchit-gandhi/whisper-jax
A JAX implementation of OpenAI's Whisper model for up to 70x speed-up on TPU for speech-to-text tasks.
lululxvi/deepxde
A Python library for scientific machine learning and physics-informed learning
google-research/scenic
JAX library for computer vision research with transformers, attention mechanisms, and vision models
iree-org/iree
A retargetable MLIR-based machine learning compiler and runtime toolkit for AI/ML developers.
google-deepmind/dm-haiku
A JAX-based neural network library for building and training deep learning models.
Stay in the loop
Get weekly updates on trending AI coding tools and projects.