Ecosyste.ms: Issues

An open API service for providing issue and pull request metadata for open source projects.

GitHub / apple/axlearn issues and pull requests

#966 - Fix incorrect number of formatting arguments

Pull Request - State: closed - Opened by changlan 1 day ago

#965 - Reduce the verbosity of variable norm summaries

Pull Request - State: closed - Opened by dunan 4 days ago

#964 - [Enhancement] Add PPO/GRPO

Issue - State: open - Opened by sbhavani 4 days ago

#962 - Sliding window support for GPU flash attention

Pull Request - State: closed - Opened by kelvin-zou 7 days ago

#961 - Skipping empty grain batches during unbatch.

Pull Request - State: closed - Opened by markblee 7 days ago

#960 - Supports loss_weights and live_targets in metrics.

Pull Request - State: closed - Opened by markblee 8 days ago

#959 - Improve gcsfuse io through setting a couple of flags

Pull Request - State: closed - Opened by RsEnts 8 days ago

#958 - SplashAttention performance tuning for v6e

Pull Request - State: closed - Opened by hanzhi713 9 days ago - 1 comment

#957 - Use env id for gcp settings

Pull Request - State: open - Opened by Ethanlm 9 days ago

#956 - Use InputDispatcher for fuji models

Pull Request - State: closed - Opened by hanzhi713 10 days ago

#955 - Add v6e PCIe overload workaround flag

Pull Request - State: closed - Opened by hanzhi713 10 days ago

#954 - Fix GCSFUSE flags by setting resource limit.

Pull Request - State: closed - Opened by RsEnts 10 days ago

#953 - Explicitly pass module outputs to metrics.

Pull Request - State: closed - Opened by markblee 11 days ago

#952 - Add v6e special meshes

Pull Request - State: closed - Opened by hanzhi713 12 days ago - 1 comment

#951 - Workaround module outputs being dropped.

Pull Request - State: closed - Opened by markblee 12 days ago

#950 - Adds logging to aux loss collection.

Pull Request - State: open - Opened by ruomingp 12 days ago

#949 - Update LoraFusedQKVLinear

Pull Request - State: closed - Opened by qdavid1 13 days ago

#948 - update jax to 0.4.37

Pull Request - State: closed - Opened by matthew-e-hopkins 13 days ago

#947 - Add link to github issue regarding kubernetes-32.0.0

Pull Request - State: closed - Opened by Ethanlm 14 days ago

#945 - J

Issue - State: closed - Opened by julianandresr199 15 days ago

#945 - J

Issue - State: closed - Opened by julianandresr199 15 days ago

#944 - Forward input keys to decoder.

Pull Request - State: closed - Opened by markblee 15 days ago

#944 - Forward input keys to decoder.

Pull Request - State: closed - Opened by markblee 15 days ago

#943 - Legacy flash remat fix

Pull Request - State: closed - Opened by hanzhi713 15 days ago

#942 - Some fixes for flash remat

Pull Request - State: closed - Opened by hanzhi713 16 days ago

#941 - Uses pytest markers instead of module skip.

Pull Request - State: open - Opened by markblee 16 days ago

#940 - Add GKE A3 Ultra support

Pull Request - State: open - Opened by samos123 16 days ago - 1 comment

#939 - Flash Attention for Neuron

Pull Request - State: open - Opened by apoorvtintin 17 days ago - 4 comments

#938 - Repeat KV heads in Flash Attention

Pull Request - State: closed - Opened by changlan 17 days ago

#937 - AOT compilation for v6e

Pull Request - State: closed - Opened by changlan 17 days ago

#936 - Adds mesh rule for a3-megagpu-8g.

Pull Request - State: closed - Opened by markblee 18 days ago

#935 - Avoid a top-level import of tokenizers.

Pull Request - State: closed - Opened by markblee 19 days ago

#934 - Makes causal lm metrics configurable.

Pull Request - State: closed - Opened by markblee 19 days ago - 1 comment

#933 - Supports flexible input partition specs.

Pull Request - State: closed - Opened by markblee 20 days ago - 1 comment

#932 - Enable GCP Workload Monitoring

Pull Request - State: open - Opened by Perseus14 21 days ago

#931 - Add ReadOptions args to _make_autoregressive_inputs

Pull Request - State: closed - Opened by RsEnts 21 days ago

#930 - AdaptiveLayerNormModulation raises ValueError, instead of assert.

Pull Request - State: closed - Opened by ds-hwang 22 days ago - 1 comment

#929 - Fix aot compilation with grain inputs.

Pull Request - State: closed - Opened by markblee 22 days ago

#928 - Add prefill hidden states as module outputs.

Pull Request - State: closed - Opened by markblee 22 days ago - 1 comment

#927 - Cache AoT compilation result

Pull Request - State: closed - Opened by hanzhi713 22 days ago

#926 - Allow external positions to be inputed in RoPE embedding layer

Pull Request - State: closed - Opened by Firenze11 23 days ago

#925 - Support remat for FlashAttention

Pull Request - State: closed - Opened by hanzhi713 24 days ago

#924 - Fix tf iter unit test

Pull Request - State: closed - Opened by hanzhi713 24 days ago

#923 - DiT now supports sequence conditions.

Pull Request - State: closed - Opened by ds-hwang 24 days ago - 1 comment

#922 - Enabled running Pallas Flash Attention on CPU.

Pull Request - State: closed - Opened by ds-hwang 24 days ago - 3 comments

#921 - [Grain] Minor Fix for Version Update

Pull Request - State: closed - Opened by zxybazh 24 days ago

#920 - Introduce `BaseAttentionBias.has_value()`.

Pull Request - State: closed - Opened by ds-hwang 25 days ago - 1 comment

#918 - Draft: Refactor JobSet for Pathways

Pull Request - State: open - Opened by jiya-zhang 28 days ago - 1 comment

#917 - fix broken apt install google-perftools

Pull Request - State: closed - Opened by samos123 28 days ago - 1 comment

#916 - TRN2 Meshes and Configurations

Pull Request - State: open - Opened by apoorvtintin 28 days ago - 9 comments

#914 - Optional `positions` support in decoder and attention layers

Pull Request - State: closed - Opened by changlan 29 days ago

#913 - Enable cudnn attention dropout

Pull Request - State: open - Opened by hanzhi713 30 days ago - 3 comments

#912 - V6e support

Pull Request - State: closed - Opened by kelvin-zou 30 days ago - 3 comments

#911 - Update lora input linear adapter output dim.

Pull Request - State: closed - Opened by JianyuWangV 30 days ago - 1 comment

#910 - Allow parallel gpu tests

Pull Request - State: closed - Opened by hanzhi713 about 1 month ago

#909 - Introduce the scale enum flag in Embedding layer for LLM embedding.

Pull Request - State: closed - Opened by ds-hwang about 1 month ago - 1 comment

#908 - Optimize TPU Flash Attention (20x XLA compilation speed-up on 32k long context)

Pull Request - State: closed - Opened by ds-hwang about 1 month ago - 7 comments

#907 - The codebook of the KmeansVectorQuantizer should be initialized with scale=1/sqrt(dim).

Pull Request - State: closed - Opened by ds-hwang about 1 month ago - 1 comment

#906 - `AdaptiveLayerNormModulation` now supports sequence conditions.

Pull Request - State: closed - Opened by ds-hwang about 1 month ago - 1 comment

#905 - Flash2 and supports cross attention and dropout

Pull Request - State: closed - Opened by hanzhi713 about 1 month ago - 1 comment

#904 - Remove version pin of typing-extensions

Pull Request - State: closed - Opened by wangkuiyi about 1 month ago - 2 comments

#903 - Fix softmax scale arg passing

Pull Request - State: closed - Opened by hanzhi713 about 2 months ago

#901 - Little clean-up in frontend.

Pull Request - State: closed - Opened by ds-hwang about 2 months ago - 1 comment

#899 - Implements FlashDecoding with Sparsity Support

Pull Request - State: closed - Opened by hanzhi713 about 2 months ago

#899 - Implements FlashDecoding with Sparsity Support

Pull Request - State: closed - Opened by hanzhi713 about 2 months ago

#898 - Special remat for Neuron

Pull Request - State: closed - Opened by apoorvtintin about 2 months ago - 17 comments

#891 - N

Issue - State: closed - Opened by julianandresr199 about 2 months ago

#890 - use "true" and "false" instead of 0 and 1

Pull Request - State: open - Opened by samos123 about 2 months ago - 1 comment

#888 - MaskFnAttentionBias._bool_value passes the same rank position tensors to mask_fn.

Pull Request - State: open - Opened by ds-hwang about 2 months ago - 1 comment

#887 - Add default compiler options for v6e

Pull Request - State: closed - Opened by samos123 about 2 months ago

#886 - [DO-NOT-MERGE] PR encompassing all changes needed to support neuron on Axlearn

Pull Request - State: closed - Opened by apoorvtintin about 2 months ago - 1 comment

#885 - Add meshes and config for TRN2/1 for Fuji models

Pull Request - State: closed - Opened by apoorvtintin about 2 months ago - 2 comments

#884 - Input batch sharding strategy BATCH

Pull Request - State: open - Opened by apoorvtintin about 2 months ago - 10 comments

#883 - Flash Attention for Neuron

Pull Request - State: closed - Opened by apoorvtintin about 2 months ago - 1 comment

#882 - Update RoformerQKVLinear to support kv_state

Pull Request - State: open - Opened by qdavid1 about 2 months ago

#881 - WIP - Support fine grained activation sharding.

Pull Request - State: open - Opened by ptoulme-aws about 2 months ago

#881 - WIP - Support fine grained activation sharding.

Pull Request - State: open - Opened by ptoulme-aws about 2 months ago

#880 - Docker: Upgrade Jax to 0.4.37

Pull Request - State: closed - Opened by samos123 about 2 months ago

#880 - Docker: Upgrade Jax to 0.4.37

Pull Request - State: open - Opened by samos123 about 2 months ago

#879 - Support v6e (Trillium)

Pull Request - State: closed - Opened by Ethanlm about 2 months ago

#879 - Support v6e (Trillium)

Pull Request - State: closed - Opened by Ethanlm about 2 months ago

#878 - [bug fix] fix issues with tool use metrics

Pull Request - State: closed - Opened by fnan about 2 months ago

#877 - Makes Bastion write scheduler history files

Pull Request - State: closed - Opened by ruomingp about 2 months ago

#877 - Makes Bastion write scheduler history files

Pull Request - State: closed - Opened by ruomingp about 2 months ago

#876 - DiT: implement init_states and extend_step for DiT transformer

Pull Request - State: closed - Opened by ds-hwang about 2 months ago - 2 comments

#876 - DiT: implement init_states and extend_step for DiT transformer

Pull Request - State: closed - Opened by ds-hwang about 2 months ago - 2 comments

#875 - pin scikit-learn

Pull Request - State: closed - Opened by fnan about 2 months ago

#875 - pin scikit-learn

Pull Request - State: closed - Opened by fnan about 2 months ago

#874 - Remove usage of jax.core and jax.interpreters

Issue - State: open - Opened by samos123 about 2 months ago - 1 comment

#874 - Remove usage of jax.core and jax.interpreters

Issue - State: open - Opened by samos123 about 2 months ago - 1 comment

#873 - Refactor attention bias/mask/segments.

Pull Request - State: closed - Opened by apghml 2 months ago - 1 comment

#873 - Refactor attention bias/mask/segments.

Pull Request - State: closed - Opened by apghml 2 months ago - 1 comment

#872 - Simplifies ScheduleResults.job_verdicts

Pull Request - State: closed - Opened by ruomingp 2 months ago - 2 comments

#871 - add tool use intent metrics

Pull Request - State: closed - Opened by fnan 2 months ago

#871 - add tool use intent metrics

Pull Request - State: closed - Opened by fnan 2 months ago