-
torch2jax Public
Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
-
tune-jax Public
Microbenchmarking hyperparameter tuning for JAX functions.
-
gemma Public
Forked from google-deepmind/gemmaOpen weights LLM from Google DeepMind.
Jupyter Notebook Apache License 2.0 UpdatedNov 22, 2024 -
jax Public
Forked from jax-ml/jaxComposable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Python Apache License 2.0 UpdatedNov 5, 2024 -
liger_kernels Public
Forked from phlippe/liger_kernelsJAX Implementation of Liger Kernels
Python UpdatedOct 29, 2024 -
entropix Public
Forked from Angelogeb/entropixEntropy Based Sampling and Parallel CoT Decoding
TypeScript Apache License 2.0 UpdatedOct 25, 2024 -
jax-triton Public
Forked from jax-ml/jax-tritonjax-triton contains integrations between JAX and OpenAI Triton
-
-
weather_landscape Public
Forked from lds133/weather_landscapeVisualizing Weather Forecasts Through Landscape Imagery
-
maxtext Public
Forked from AI-Hypercomputer/maxtextA simple, performant and scalable Jax LLM!
Python Apache License 2.0 UpdatedSep 25, 2024 -
ml-auto-solutions Public
Forked from GoogleCloudPlatform/ml-auto-solutionsA simplified and automated orchestration workflow to perform ML end-to-end (E2E) model tests and benchmarking on Cloud VMs across different frameworks.
Python Apache License 2.0 UpdatedSep 16, 2024 -
-
llama3 Public
Forked from meta-llama/llama3The official Meta Llama 3 GitHub site
Python Other UpdatedAug 12, 2024 -
ray Public
Forked from ray-project/rayRay is a unified framework for scaling AI and Python applications. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.
Python Apache License 2.0 UpdatedAug 10, 2024 -
jaxfi-JAXFriendlyInterface Public
Friendly Interface to JAX, `jaxfi` simplifies JAX interface to replicate PyTorch's ease of use.
-
orbax Public
Forked from google/orbaxOrbax provides common utility libraries for JAX users.
Python Apache License 2.0 UpdatedAug 9, 2024 -
-
-
transformers Public
Forked from huggingface/transformers🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
Python Apache License 2.0 UpdatedMay 12, 2024 -
MTR Public
Forked from sshaoshuai/MTRMTR: Motion Transformer with Global Intention Localization and Local Movement Refinement, NeurIPS 2022.
Python Apache License 2.0 UpdatedMay 9, 2024 -
zsh-vi-mode Public
Forked from jeffreytse/zsh-vi-mode💻 A better and friendly vi(vim) mode plugin for ZSH.
Shell MIT License UpdatedMar 25, 2024 -
SpAutoDiff.jl Public
Full [Sp]arse Graph [AutoDiff]erentiation Package for Julia
-
bitsandbytes Public
Forked from bitsandbytes-foundation/bitsandbytesAccessible large language models via k-bit quantization for PyTorch.
Python MIT License UpdatedFeb 28, 2024 -
mpcjax Public
A JAX library for Sequential Convex Programming (SCP) Model Predictive Control (MPC) path planning
-
pyjulia Public
Forked from JuliaPy/pyjuliapython interface to julia
Python MIT License UpdatedJan 17, 2024 -
-
guest_container_tool Public
This basic tool allows you to give others access to your computational resources inside an isolated container, without access to your server or other containers.
-
semantic_bibtool Public
Automatic generation of bibs from a single title or a list of titles with semantic scholar API.
Python MIT License UpdatedNov 10, 2023 -
icon-db Public
Forked from marmooo/icon-dbIcon Database for Icon Search. 160+ icon sets with 210,000+ icons.
UpdatedNov 5, 2023 -