jax
60 proyectos comparten este topic de GitHub
Las líneas conectan a los miembros que están mediblemente relacionados entre sí. El tamaño de los puntos refleja las estrellas.
🧬 Miembros
keras
Deep Learning for humans
★ 64.2k
d2l-en
Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities…
★ 29.3k
best-of-ml-python
🏆 A ranked list of awesome machine learning Python libraries. Updated weekly.
★ 23.7k
OpenMythos
A theoretical reconstruction of the Claude Mythos architecture, built from first principles using the…
★ 14.8k
wandb
The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from…
★ 11.2k
einops
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
★ 9.6k
autograd
Efficiently computes derivatives of NumPy code.
★ 7.5k
maths-cs-ai-compendium
Become a cracked AI/ML researcher/engineer with this unconventional textbook covering maths, computing, and…
★ 7.2k
pyprobml
Python code for "Probabilistic Machine learning" book by Kevin Murphy
★ 7.1k
interviews.ai
It is my belief that you, the postgraduate students and job-seekers for whom the book is primarily meant will…
★ 4.9k
whisper-jax
JAX implementation of OpenAI's Whisper model for up to 70x speed-up on TPU.
★ 4.7k
datasets
TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...
★ 4.6k
deepxde
A library for scientific machine learning and physics-informed learning
★ 4.3k
iree
A retargetable MLIR-based machine learning compiler and runtime toolkit.
★ 3.9k
scenic
Scenic: A Jax Library for Computer Vision Research and Beyond
★ 3.8k
dreamerv3
Mastering Diverse Domains through World Models
★ 3.6k
TransformerEngine
A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit and 4-bit floating point…
★ 3.5k
pennylane
PennyLane is an open-source quantum software platform for quantum computing, quantum machine learning, and…
★ 3.4k
dm-haiku
JAX-based neural network library
★ 3.3k
brax
Massively parallel rigidbody physics simulation on accelerator hardware.
★ 3.2k
uvadlc_notebooks
Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam…
★ 3.2k
foolbox
A Python toolbox to create adversarial examples that fool neural networks in PyTorch, TensorFlow, and JAX
★ 3k
equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
★ 2.9k
thinc
🔮 A refreshing functional take on deep learning, compatible with your favorite libraries
★ 2.9k
mctx
Monte Carlo tree search in JAX
★ 2.6k
EasyLM
Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning,…
★ 2.5k
maxtext
A simple, performant, and scalable Jax LLM!
★ 2.4k
axlearn
An Extensible Deep Learning Library
★ 2.4k
trainer
Distributed AI Model Training and LLM Fine-Tuning on Kubernetes
★ 2.2k
awesome-jax
JAX - A curated list of resources https://github.com/google/jax
★ 2.1k
diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable.…
★ 2.1k
deepmd-kit
A deep learning package for many-body potential energy representation and molecular dynamics
★ 2k
penzai
A JAX research toolkit for building, editing, and visualizing neural networks.
★ 1.9k
score_sde
Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)
★ 1.8k
magenta-realtime
Magenta RealTime 2: An Open-Weights Live Music Model
★ 1.7k
katib
Automated Machine Learning on Kubernetes
★ 1.7k
tensorly
TensorLy: Tensor Learning in Python.
★ 1.7k
PDEBench
PDEBench: An Extensive Benchmark for Scientific Machine Learning
★ 1.2k
Treasure-of-Transformers
💁 Awesome Treasure of Transformers Models for Natural Language processing contains papers, videos, blogs,…
★ 1.2k
purejaxrl
Really Fast End-to-End Jax RL Implementations
★ 1.1k
skrl
Modular Reinforcement Learning (RL) library (implemented in PyTorch, JAX, and NVIDIA Warp) with support for…
★ 1.1k
Mava
🦁 A research-friendly codebase for fast experimentation of multi-agent reinforcement learning in JAX
★ 924
jumanji
🕹️ A diverse suite of scalable reinforcement learning environments in JAX
★ 854
jaxrl
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
★ 757
Machine-Learning-Guide
Machine learning Guide. Learn all about Machine Learning Tools, Libraries, Frameworks, Large Language Models…
★ 707
fast-plate-ocr
Lightweight & fast OCR models for license plate text recognition.
★ 644
pgx
♟️ Vectorized RL game environments in JAX
★ 634
paxml
Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and…
★ 556
Quadruped-PyMPC
A model predictive controller for quadruped robots based on the single rigid body model and written in…
★ 493
JetStream
JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs…
★ 451
GradCache
Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Constraint
★ 444
dm_pix
PIX is an image processing library in JAX, for JAX.
★ 442
Stoix
🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX…
★ 416
whisper-finetune
Fine-tune and evaluate Whisper models for Automatic Speech Recognition (ASR) on custom datasets or datasets…
★ 365
jax-am
Additive manufacturing simulation with JAX.
★ 360
evorl
EvoRL is a fully GPU-accelerated framework for Evolutionary Reinforcement Learning, implemented with JAX. It…
★ 343
xland-minigrid
JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid 🏎️
★ 342
jaxlie
Rigid transforms + Lie groups for JAX
★ 337
nanodl
JAX library for training sub-4B foundation models for edge
★ 306
gemma_penzai
A JAX Research Toolkit for Visualizing, Manipulating, and Understanding Gemma Models with Multi-modal Support…
★ 98
🔗 Familias relacionadas
Medido a partir de los temas de GitHub compartidos por ambos proyectos, ponderado por cuán raros son cada uno de los temas.